Python字典理解


386

是否可以在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

不过,我得到一个SyntaxErrorfor

另外(我不需要这部分,只是想知道),您能否将字典的键设置为一堆不同的值,例如:

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)]

这也引发了SyntaxErrorfor


3
有关将来的读者的信息:NumPy数组允许您将多个元素设置为单个值或值列表,这是您尝试执行的方式。尽管如果您还没有理由使用NumPy,则仅为此功能可能不值得。
David Z

Answers:


520

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)将新值合并到旧字典中。


15
FWIW,dict.update也可以像dict构造函数一样接受可迭代的键-值对
mgilson 2013年

6
请注意,如果要创建一个所有值都相同的字典,请使用dict.fromkeys()。因此,要将所有值设置为True,请使用dict.fromkeys(range(5), True)。注意,该值不会被复制,因此,当您具有可变的值时,可能要避免这种情况。它将在所有密钥之间共享。
马丁·彼得斯

2
注意:键也可以是方法的结果:{ n*2 : n for n in range(3) } => {0: 0, 2: 1, 4: 2}。两者都可以在同一表达式中完成:{ n*2 : n*3 for n in range(3) } => { 0: 0, 2: 3, 4: 6 }
Zaaier

151

您可以使用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]} !!!

如果您实际上不需要初始化所有键,那么a defaultdict也可能很有用:

from collections import defaultdict
d = defaultdict(True)

要回答第二部分,您需要的是dict理解:

{k: k for k in range(10)}

您可能不应该这样做,但也可以创建一个子类,如果您重写dict它,该子类的工作原理类似于a :defaultdict__missing__

>>> 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)
{}

请注意,在情况下d = defaultdict(lambda: True),不需要lambda,因为True是(或不应)是可变的。
rhbvkleef

28
>>> {i:i for i in range(1, 11)}
{1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9, 10: 10}

22

我真的很喜欢@mgilson注释,因为如果您有两个可迭代项,一个与键相对应,另一个与值相对应,则还可以执行以下操作。

keys = ['a', 'b', 'c']
values = [1, 2, 3]
d = dict(zip(keys, values))

给予

d = {'a':1,'b':2,'c':3}


4
使用Dict理解:{i:j for i in keys for j in values}
LoMaPh

11

在元组列表上使用dict(),此解决方案将允许您在每个列表中具有任意值,只要它们的长度相同

i_s = range(1, 11)
x_s = range(1, 11)
# x_s = range(11, 1, -1) # Also works
d = dict([(i_s[index], x_s[index], ) for index in range(len(i_s))])

12
作为旁注,这与d = dict(zip(i_s,x_s))
mgilson

10

考虑以下示例,该示例使用字典理解来计算列表中单词的出现

my_list = ['hello', 'hi', 'hello', 'today', 'morning', 'again', 'hello']
my_dict = {k:my_list.count(k) for k in my_list}
print(my_dict)

结果是

{'again': 1, 'hi': 1, 'hello': 3, 'today': 1, 'morning': 1}

这很有趣,尽管不是最有效的,因为您将多次计数“ hello”之类的键
FuriousGeorge

7

列表理解的主要目的是在不更改或破坏原始列表的基础上创建基于另一个列表的新列表。

而不是写作

    l = []
    for n in range(1, 11):
        l.append(n)

要么

    l = [n for n in range(1, 11)]

你应该只写

    l = range(1, 11)

在两个最重要的代码块中,您将创建一个新列表,对其进行迭代并仅返回每个元素。这只是创建列表副本的一种昂贵方法。

要获得一个新的字典,并且根据另一个字典将所有键设置为相同的值,请执行以下操作:

    old_dict = {'a': 1, 'c': 3, 'b': 2}
    new_dict = { key:'your value here' for key in old_dict.keys()}

您收到SyntaxError的原因是,在编写时

    d = {}
    d[i for i in range(1, 11)] = True

您基本上是在说:“将我的密钥'i for range(1,11)'中的i设置为True”和“ i for i in range(1,11)中的”)不是有效的密钥,这只是语法错误。如果将dicts支持的列表作为键,您将执行以下操作

    d[[i for i in range(1, 11)]] = True

并不是

    d[i for i in range(1, 11)] = True

但是列表不可散列,因此您不能将它们用作字典键。


By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.