是否有可能在Python中创建一个字典理解(用于键)?
如果没有列表推导式,你可以这样使用:
l = []
for n in range(1, 11):
l.append(n)
我们可以将其缩短为一个列表推导式:l = [n for n in range(1,11)]。
但是,假设我想将字典的键值设置为相同的值。
我可以:
d = {}
for n in range(1, 11):
d[n] = True # same value for each
我试过了:
d = {}
d[i for i in range(1, 11)] = True
但是,我在for上得到了一个SyntaxError。
此外(我不需要这部分,但只是想知道),你可以将字典的键设置为一堆不同的值,像这样:
d = {}
for n in range(1, 11):
d[n] = n
这在字典理解中是可能的吗?
d = {}
d[i for i in range(1, 11)] = [x for x in range(1, 11)]
这还会在for上引发SyntaxError。
你可以使用dict.fromkeys类方法…
>>> dict.fromkeys(range(5), True)
{0: True, 1: True, 2: True, 3: True, 4: True}
这是创建所有键都映射到相同值的字典的最快方法。
但是不要对可变对象使用这个:
d = dict.fromkeys(range(5), [])
# {0: [], 1: [], 2: [], 3: [], 4: []}
d[1].append(2)
# {0: [2], 1: [2], 2: [2], 3: [2], 4: [2]} !!!
如果你实际上不需要初始化所有的键,一个defaultdict可能也很有用:
from collections import defaultdict
d = defaultdict(True)
要回答第二部分,你需要的是对词典的理解:
{k: k for k in range(10)}
你可能不应该这样做,但你也可以创建一个dict的子类,如果你重写__missing__,它的工作方式有点像defaultdict:
>>> class KeyDict(dict):
... def __missing__(self, key):
... #self[key] = key # Maybe add this also?
... return key
...
>>> d = KeyDict()
>>> d[1]
1
>>> d[2]
2
>>> d[3]
3
>>> print(d)
{}
Python 2.7+中有字典推导式,但它们不像您尝试的那样工作。像列表理解一样,它们创建了一个新字典;不能使用它们向现有字典添加键。此外,您必须指定键和值,当然,如果您愿意,您可以指定一个虚拟值。
>>> d = {n: n**2 for n in range(5)}
>>> print d
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16}
如果你想把它们都设置为True:
>>> d = {n: True for n in range(5)}
>>> print d
{0: True, 1: True, 2: True, 3: True, 4: True}
您所要求的似乎是一种在现有字典上一次性设置多个键的方法。没有捷径可走。您可以像前面演示的那样进行循环,也可以使用字典推导来创建具有新值的新字典,然后执行oldDict.update(newDict)将新值合并到旧字典中。
你可以使用dict.fromkeys类方法…
>>> dict.fromkeys(range(5), True)
{0: True, 1: True, 2: True, 3: True, 4: True}
这是创建所有键都映射到相同值的字典的最快方法。
但是不要对可变对象使用这个:
d = dict.fromkeys(range(5), [])
# {0: [], 1: [], 2: [], 3: [], 4: []}
d[1].append(2)
# {0: [2], 1: [2], 2: [2], 3: [2], 4: [2]} !!!
如果你实际上不需要初始化所有的键,一个defaultdict可能也很有用:
from collections import defaultdict
d = defaultdict(True)
要回答第二部分,你需要的是对词典的理解:
{k: k for k in range(10)}
你可能不应该这样做,但你也可以创建一个dict的子类,如果你重写__missing__,它的工作方式有点像defaultdict:
>>> class KeyDict(dict):
... def __missing__(self, key):
... #self[key] = key # Maybe add this also?
... return key
...
>>> d = KeyDict()
>>> d[1]
1
>>> d[2]
2
>>> d[3]
3
>>> print(d)
{}