Python字典理解

Python Dictionary Comprehension

是否可以用python(键)创建字典理解?

如果不理解列表,您可以使用如下内容:

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

我们可以把它缩短为一个列表理解:l = [n for n in range(1, 11)]

但是,假设我想将字典的键设置为相同的值。我能做到:

1
2
3
d = {}
for n in range(1, 11):
     d[n] = True # same value for each

我试过了:

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

但是,我在for上得到了一个SyntaxError

另外(我不需要这部分,只是想知道),你能把字典的键设置成一组不同的值吗,比如:

1
2
3
d = {}
for n in range(1, 11):
    d[n] = n

这可以用字典来理解吗?

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

这也使SyntaxErrorfor上出现。


在python 2.7+中有字典的理解,但是它们并不像您所尝试的那样工作。就像列表理解一样,它们创建了一个新字典;您不能使用它们向现有字典添加键。此外,还必须指定键和值,当然,如果愿意,也可以指定一个虚拟值。

1
2
3
>>> d = {n: n**2 for n in range(5)}
>>> print d
{0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

如果要将它们全部设置为真:

1
2
3
>>> d = {n: True for n in range(5)}
>>> print d
{0: True, 1: True, 2: True, 3: True, 4: True}

您似乎需要的是一种在现有字典上同时设置多个键的方法。没有直接的捷径。您可以像已经显示的那样循环,也可以使用字典理解创建一个具有新值的新dict,然后执行oldDict.update(newDict)将新值合并到旧dict中。


您可以使用dict.fromkeys类方法…

1
2
>>> dict.fromkeys(range(5), True)
{0: True, 1: True, 2: True, 3: True, 4: True}

这是创建所有键映射到相同值的字典的最快方法。

但不要将其用于可变对象:

1
2
3
4
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可能也很有用:

1
2
from collections import defaultdict
d = defaultdict(True)

要回答第二部分,听写理解正是你需要的:

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

您可能不应该这样做,但也可以创建dict的子类,如果您重写__missing__,它的工作方式有点像defaultdict

1
2
3
4
5
6
7
8
9
10
11
12
13
14
>>> 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)
{}


1
2
>>> {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}

我真的很喜欢@mgilson注释,因为如果您有两个iterables,一个对应于键,另一个对应于值,那么您也可以执行以下操作。

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

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


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

1
2
3
4
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))])


考虑这个使用字典理解计算列表中单词出现次数的例子。

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

结果是

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


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

而不是写作

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

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

你应该只写

1
    l = range(1, 11)

在这两个顶级代码块中,您创建了一个新的列表,遍历它并返回每个元素。这只是创建列表副本的昂贵方法。

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

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

你收到一个语法错误,因为当你写

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

您基本上是说:"将我的键‘I for I in range(1,11)’设置为"真",而‘I for I in range(1,11)’不是有效的键,这只是一个语法错误。如果dicts支持将列表作为键,您可以执行如下操作

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

而不是

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

但是列表是不可哈希的,所以不能将它们用作dict键。


你不能这样散列一个列表。试试这个,它使用元组

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