我喜欢 Python 列表理解语法。
它也可以用来创建字典吗?例如,通过迭代键和值对:
mydict = {(k,v) for (k,v) in blah blah blah} # doesn't work
我喜欢 Python 列表理解语法。
它也可以用来创建字典吗?例如,通过迭代键和值对:
mydict = {(k,v) for (k,v) in blah blah blah} # doesn't work
使用dict 理解(Python 2.7 及更高版本):
{key: value for (key, value) in iterable}
或者,对于更简单的情况或更早版本的 Python,请使用dict
构造函数,例如:
pairs = [('a', 1), ('b', 2)]
dict(pairs) #=> {'a': 1, 'b': 2}
dict([(k, v+1) for k, v in pairs]) #=> {'a': 2, 'b': 3}
给定单独的键和值数组,使用dict
构造函数zip
:
keys = ['a', 'b']
values = [1, 2]
dict(zip(keys, values)) #=> {'a': 1, 'b': 2}
2) "zip'ped" from two separate iterables of keys/vals
dict(zip(list_of_keys, list_of_values))
在 Python 3 和 Python 2.7+ 中,字典推导如下所示:
d = {k:v for k, v in iterable}
对于 Python 2.6 或更早版本,请参阅fortran 的答案。
事实上,如果它已经理解某种映射,你甚至不需要遍历 iterable,dict 构造函数会优雅地为你做这件事:
>>> 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}
在 Python 中创建具有列表理解的字典
我喜欢 Python 列表理解语法。
它也可以用来创建字典吗?例如,通过迭代键和值对:
mydict = {(k,v) for (k,v) in blah blah blah}
您正在寻找短语“dict理解” - 它实际上是:
mydict = {k: v for k, v in iterable}
假设blah blah blah
是两个元组的可迭代 - 你是如此接近。让我们创建一些这样的“废话”:
blahs = [('blah0', 'blah'), ('blah1', 'blah'), ('blah2', 'blah'), ('blah3', 'blah')]
现在这里的语法是映射部分。是什么使它成为dict
理解而不是set
理解(这是您的伪代码近似的)是冒号,:
如下所示:
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 整除,以便在映射键和值之前过滤掉数据。
在 Python 2.7 中,它是这样的:
>>> list1, list2 = ['a', 'b', 'c'], [1,2,3]
>>> dict( zip( list1, list2))
{'a': 1, 'c': 3, 'b': 2}
压缩它们!
Python 版本 >= 2.7,请执行以下操作:
d = {i: True for i in [1,2,3]}
Python 版本 < 2.7 (RIP,2010 年 7 月 3 日 - 2019 年 12 月 31 日),请执行以下操作:
d = dict((i,True) for i in [1,2,3])
要添加到@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)}
这是另一个使用字典理解创建字典的示例:
我在这里要做的是创建一个字母字典,其中每一对;是英文字母及其在英文字母表中的对应位置
>>> 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 来获取列表中的字母列表及其索引,并交换字母和索引以生成字典的键值对
希望它能给你一个字典编译的好主意,并鼓励你更频繁地使用它来使你的代码更紧凑
只是举另一个例子。假设您有以下列表:
nums = [4,2,2,1,3]
你想把它变成一个字典,其中键是索引,值是列表中的元素。您可以使用以下代码行执行此操作:
{index:nums[index] for index in range(0,len(nums))}
此代码将使用列表推导为具有不同值的多个列表创建字典,这些列表可用于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
将传递给以n
匹配vals
每个key
与其列表
试试这个,
def get_dic_from_two_lists(keys, values):
return { keys[i] : values[i] for i in range(len(keys)) }
假设我们有两个列表国家和首都
country = ['India', 'Pakistan', 'China']
capital = ['New Delhi', 'Islamabad', 'Beijing']
然后从两个列表创建字典:
print get_dic_from_two_lists(country, capital)
输出是这样的,
{'Pakistan': 'Islamabad', 'China': 'Beijing', 'India': 'New Delhi'}
>>> {k: v**3 for (k, v) in zip(string.ascii_lowercase, range(26))}
Python 支持字典推导式,它允许您在运行时使用类似简洁的语法来表达字典的创建。
字典推导采用 {key: value for (key, value) in iterable} 的形式。这种语法是在 Python 3 中引入的,并且向后移植到 Python 2.7,因此无论您安装了哪个版本的 Python,您都应该能够使用它。
一个典型的例子是获取两个列表并创建一个字典,其中第一个列表中每个位置的项目成为键,第二个列表中相应位置的项目成为值。
这个理解中使用的 zip 函数返回一个元组迭代器,其中元组中的每个元素都取自每个输入迭代中的相同位置。在上面的示例中,返回的迭代器包含元组 (“a”, 1)、(“b”, 2) 等。
输出:
{'i': 512, 'e': 64, 'o': 2744, 'h': 343, 'l': 1331, 's': 5832, 'b': 1, 'w': 10648, 'c': 8, 'x': 12167, 'y': 13824, 't': 6859, 'p': 3375, 'd': 27, 'j': 729, 'a': 0, 'z': 15625, 'f': 125, 'q': 4096, 'u': 8000, 'n': 2197, 'm': 1728, 'r': 4913, 'k': 1000, 'g': 216, 'v': 9261}
添加到@Ekhtiar answer,如果您想dict
从查找list
,可以使用:
names = ['a', 'b', 'd', 'f', 'c']
names_to_id = {v:k for k, v in enumerate(names)}
# {'a': 0, 'b': 1, 'c': 2, 'd': 3, 'f': 4}
或者在极少数情况下要过滤重复项,set
请先使用(最好在数字列表中):
names = ['a', 'b', 'd', 'f', 'd', 'c']
sorted_list = list(set(names))
sorted_list.sort()
names_to_id = {v:k for k, v in enumerate(sorted_list)}
# {'a': 0, 'b': 1, 'c': 2, 'd': 3, 'f': 4}
names = [1,2,5,5,6,2,1]
names_to_id = {v:k for k, v in enumerate(set(names))}
# {1: 0, 2: 1, 5: 2, 6: 3}
是的,这是可能的。在python中,List、Set、Dictionary等都可以使用Comprehension,可以这样写
mydict = {k:v for (k,v) in blah}
使用条件语句和循环进行字典理解的另一个详细示例:
parents = [father, mother]
parents = {parent:1 - P["mutation"] if parent in two_genes else 0.5 if parent in one_gene else P["mutation"] for parent in parents}
假设 blah blah blah 是一个二元组列表:
让我们看看两种方法:
# method 1
>>> lst = [('a', 2), ('b', 4), ('c', 6)]
>>> dict(lst)
{'a': 2, 'b': 4, 'c': 6}
# method 2
>>> lst = [('a', 2), ('b', 4), ('c', 6)]
>>> d = {k:v for k, v in lst}
>>> d
{'a': 2, 'b': 4, 'c': 6}
这种方法使用 for 循环在给定日期上使用迭代。
Syntax: {key: value for (key, value) in data}
例如:
# create a list comprehension with country and code:
Country_code = [('China', 86), ('USA', 1),
('Ghana', 233), ('Uk', 44)]
# use iterable method to show results
{key: value for (key, value) in Country_code}
您可以为每一对创建一个新的字典并将其与前一个字典合并:
reduce(lambda p, q: {**p, **{q[0]: q[1]}}, bla bla bla, {})
显然,这种方法需要reduce
从functools
.