使用推导式创建一个字典

1564

我能使用列表推导式语法创建一个字典吗?

例如,迭代键值对:

d = {... for k, v in zip(keys, values)}

1
相关:collections.Counter是一种专门用于计数的字典类型:使用字典计算列表中的项目数 - smci
你的例子中的 blah bla 是误导性的,你可能指的是一个元组列表,但它也可以是一个普通的 Python 列表。因此,前几个高票答案都涉及由元组列表构成的字典,而 这个 答案则不是。 - Timo
2
如果您没有使用 zip(...) 而是使用单独的字典,那么您需要执行类似于 d = {... for (k, v) in other_dict.items()} 的操作。 - Miguel Vacas
17个回答

2308
使用字典推导式(Python 2.7及更高版本):
{key: value for key, value in zip(keys, values)}

或者,使用dict构造函数:
pairs = [('a', 1), ('b', 2)]
dict(pairs)                          # → {'a': 1, 'b': 2}
dict((k, v + 10) for k, v in pairs)  # → {'a': 11, 'b': 12}

给定键和值的分开列表,使用dict构造函数和zip
keys = ['a', 'b']
values = [1, 2]
dict(zip(keys, values))              # → {'a': 1, 'b': 2}

1
如果我有一个单词列表['cat','dog','cat'],并且我想创建一个字典,其中键是单词,值是计数,是否有一种简洁高效的语法可以实现? - cryanbhu
1
@cryanbhu 如果你的意思是要计算列表中给定元素的重复次数,那么在collections包中有一个Counter类:https://docs.python.org/2/library/collections.html#collections.Counter - fortran

293

在Python3和Python2.7+中,字典推导式的写法如下:

d = {k:v for k, v in iterable}

对于Python 2.6或更早版本,请查看fortran的回答


73

实际上,如果已经包含某种映射关系,您甚至不需要对可迭代对象进行迭代,因为字典构造函数会为您优雅地完成这个过程:

>>> ts = [(1, 2), (3, 4), (5, 6)]
>>> dict(ts)
{1: 2, 3: 4, 5: 6}
>>> gen = ((i, i+1) for i in range(1, 6, 2))
>>> gen
<generator object <genexpr> at 0xb7201c5c>
>>> dict(gen)
{1: 2, 3: 4, 5: 6}

53

使用Python中的列表推导式创建字典

我喜欢Python中列表推导式的语法。

它能否也用于创建字典呢?例如,通过遍历键和值的对:

mydict = {(k,v) for (k,v) in blah blah blah}
您正在寻找“字典推导式”这个短语 - 实际上是:

dict comprehension
mydict = {k: v for k, v in iterable}

假设blah blah blah是一个由二元组组成的可迭代对象——你已经很接近了。让我们创建一些这样的“blahs”:

blahs = [('blah0', 'blah'), ('blah1', 'blah'), ('blah2', 'blah'), ('blah3', 'blah')]

字典推导式语法:

现在,这里的语法是映射部分。使得这个字典推导式不同于一个集合推导式(你的伪代码近似),是冒号:,像下面这样:

mydict = {k: v for k, v in blahs}

我们可以看到它已经生效,并且应该保留插入顺序,Python 3.7版本即可实现:

>>> mydict
{'blah0': 'blah', 'blah1': 'blah', 'blah2': 'blah', 'blah3': 'blah'}

在 Python 2 及到 3.6 版本之前,顺序不被保证:

>>> mydict
{'blah0': 'blah', 'blah1': 'blah', 'blah3': 'blah', 'blah2': 'blah'}

添加过滤器:

所有推导式都包含一个映射组件和一个筛选组件,您可以使用任意表达式来提供它们。

因此,您可以在结尾处添加一个过滤器部分:

>>> mydict = {k: v for k, v in blahs if not int(k[-1]) % 2}
>>> mydict
{'blah0': 'blah', 'blah2': 'blah'}

在这里,我们只是测试最后一个字符是否可被2整除,以便在映射键和值之前过滤数据。


44
在Python 2.7中,它是这样的:
>>> list1, list2 = ['a', 'b', 'c'], [1,2,3]
>>> dict( zip( list1, list2))
{'a': 1, 'c': 3, 'b': 2}

把它们压缩起来


12
这完全没有回答问题。 - Jean-François Corbett

33

如果Python版本大于等于2.7,请执行以下操作:

d = {i: True for i in [1,2,3]}

如果使用的是 Python 版本低于 2.7(已停止支持, 2010年7月3日 - 2019年12月31日),请按以下步骤操作:

d = dict((i,True) for i in [1,2,3])

1
我所需要的:从普通列表创建字典。 - Timo

26
补充一下@fortran的回答,如果你想要遍历一个键列表key_list以及一个值列表value_list:
d = dict((key, value) for (key, value) in zip(key_list, value_list))

或者

d = {key: value for (key, value) in zip(key_list, value_list)}

7

再举一个例子。假设你有以下列表:

nums = [4,2,2,1,3]

如果你想将一个列表转换成字典,其中键是索引,值是列表中的元素。你可以使用以下代码实现:

{index:nums[index] for index in range(0,len(nums))}

OP的问题指定了“例如,通过迭代键和值对”,因此大多数答案都集中在这方面。然而,我点赞这个答案是因为它是处理列表到字典用例的好例子。 - Subfuzion

6
这是使用字典推导式创建字典的另一个示例:
我在这里尝试做的是创建一个字母表字典,其中每一对都是英文字母和其在英文字母表中相应的位置。
>>> import string
>>> dict1 = {value: (int(key) + 1) for key, value in 
enumerate(list(string.ascii_lowercase))}
>>> dict1
{'a': 1, 'c': 3, 'b': 2, 'e': 5, 'd': 4, 'g': 7, 'f': 6, 'i': 9, 'h': 8, 
'k': 11, 'j': 10, 'm': 13, 'l': 12, 'o': 15, 'n': 14, 'q': 17, 'p': 16, 's': 
19, 'r': 18, 'u': 21, 't': 20, 'w': 23, 'v': 22, 'y': 25, 'x': 24, 'z': 26}
>>> 

注意这里使用了enumerate来获取字母列表及其在列表中的索引,并交换字母和索引以生成字典的键值对。
希望这能给你一个关于字典推导的好主意,并鼓励你更频繁地使用它来使你的代码更加简洁。

1
不错的答案 - 简化版:d = {k: v+1 for v, k in enumerate(string.ascii_lowercase)} - johnnydrama

6
这段代码将使用列表推导式创建包含多个具有不同值的列表的字典,可用于 pd.DataFrame()
#Multiple lists 
model=['A', 'B', 'C', 'D']
launched=[1983,1984,1984,1984]
discontinued=[1986, 1985, 1984, 1986]

#Dictionary with list comprehension
keys=['model','launched','discontinued']
vals=[model, launched,discontinued]
data = {key:vals[n] for n, key in enumerate(keys)}

#Convert dict to dataframe
df=pd.DataFrame(data)
display(df)
enumerate将会传递nvals,以匹配每个key与其列表。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接