在python中创建具有列表理解功能的字典


Create a dictionary with list comprehension in Python

我喜欢Python列表理解语法。

它也可以用来创建字典吗？例如，通过对键和值进行迭代：

1
mydict = {(k,v) for (k,v) in blah blah blah}  # doesn't work
相关讨论

在python 2.6及更早版本中，dict构造函数可以接收一个iterable的键/值对：

1
d = dict((key, value) for (key, value) in iterable)

从python 2.7和3开始，您可以直接使用dict理解语法：

1
d = {key: value for (key, value) in iterable}

当然，您可以以任何方式使用iterable(元组和列表文本、生成器理解、列表理解、生成器函数、函数组合…感觉有创意)只要每个元素都是由两个元素组成的不可重复的自身：

1
2
3
4
5
6
d = {value: foo(value) for value in sequence if bar(value)}

def key_value_gen(k):
   yield chr(k+65)
   yield chr((k+13)%26+65)
d = dict(map(key_value_gen, range(26)))

额外好处：如果您想循环访问键key_list的列表以及值value_list的列表：

1
d = dict((key, value) for (key, value) in zip(key_list, value_list))

或

1
d = {(key, value) for (key, value) in zip(key_list, value_list)}
相关讨论

在python 3和python 2.7+中，字典的理解如下：

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

对于Python2.6或更早版本，请参见Fortran的答案。

事实上，如果ITerable已经理解了某种映射，那么您甚至不需要对它进行迭代，dict构造函数会为您优雅地执行它：

1
2
3
4
5
6
7
8
>>> 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 2.7中，它是这样的：

1
2
3
>>> list1, list2 = ['a', 'b', 'c'], [1,2,3]
>>> dict( zip( list1, list2))
{'a': 1, 'c': 3, 'b': 2}

拉链！

相关讨论

要添加到@fortran的答案中，如果您想循环访问键key_list的列表以及值value_list的列表：

1
d = dict((key, value) for (key, value) in zip(key_list, value_list))

或

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

Create a dictionary with list comprehension in Python

I like the Python list comprehension syntax.

Can it be used to create dictionaries too? For example, by iterating
over pairs of keys and values:

1
mydict = {(k,v) for (k,v) in blah blah blah}

你在找"听写理解"这个短语，实际上是：

1
  mydict = {k:v  for k,v in iterable}

这几乎是有效的，除了废话：

1
2
mydict = {(k,v) for (k,v) in blah blah blah}
                                 ^^^^^^--invalid syntax

假设blah blah blah是两个元组中的一个不可数-你离得太近了。让我们创建一些这样的"blahs"：

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

听写理解语法：

这里的语法是映射部分。这是什么使它成为dict理解而不是set理解(这是伪代码近似值)的冒号，:如下：

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

我们看到它起作用了，并且应该保留从Python3.7开始的插入顺序：

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

在python 2和3.6之前，订单没有得到保证：

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

添加筛选器：

所有理解都有一个映射组件和一个过滤组件，您可以提供任意表达式。

因此，可以在末尾添加过滤器部件：

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

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

对于python版本<2.7，请执行以下操作：

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

对于python版本>=2.7，请执行以下操作：

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

下面是使用听写理解创建词典的另一个示例：

我在这里要做的是创建一个字母词典，其中每对字母都是英文字母及其在英文字母表中的对应位置

1
2
3
4
5
6
7
8
>>> 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 here获取列表中字母及其索引的列表，并交换字母和索引以生成字典的键值对。

希望它能为您提供一个好的字典comp概念，并鼓励您更频繁地使用它来压缩代码。

相关讨论

试试这个，

1
2
def get_dic_from_two_lists(keys, values):
    return { keys[i] : values[i] for i in range(len(keys)) }

假设我们有两个国家和资本清单

1
2
country = ['India', 'Pakistan', 'China']
capital = ['New Delhi', 'Islamabad', 'Beijing']

然后从两个列表中创建字典：

1
print get_dic_from_two_lists(country, capital)

输出是这样的，

1
{'Pakistan': 'Islamabad', 'China': 'Beijing', 'India': 'New Delhi'}
相关讨论

1
>>> {k: v**3 for (k, v) in zip(string.ascii_lowercase, range(26))}

python支持dict理解，它允许您使用类似简洁的语法在运行时表示字典的创建。

字典理解采用iterable中的形式key:value for(key，value)。这种语法是在python 3中引入的，并且可以向后移植到python 2.7，因此无论安装了哪个版本的python，您都应该能够使用它。

一个典型的例子是使用两个列表，并创建一个字典，其中第一个列表中每个位置的项都成为键，第二个列表中相应位置的项成为值。

此理解中使用的zip函数返回元组的迭代器，其中元组中的每个元素在每个输入iterables中都来自相同的位置。在上面的示例中，返回的迭代器包含元组("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

再举一个例子。假设您有以下列表：

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

你想把它变成一个dict，其中键是索引，值是列表中的元素。您可以使用以下代码行执行此操作：

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

此代码将使用列表理解为具有不同值的多个列表创建字典，这些值可用于pd.DataFrame()。

1
2
3
4
5
6
7
8
9
#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)}

enumerate将把n传给vals以使每个key与其列表相匹配。

相关讨论