Python 创建列表字典

问题描述：

我想创建一个以列表为值的字典。例如：

{
  1: ['1'],
  2: ['1','2'],
  3: ['2']
}

如果我这么做：

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2): 
        d[j].append(i)

我得到了一个 KeyError，因为 d[...] 不是列表。在这种情况下，我可以在 a 的赋值后添加以下代码来初始化字典。

for x in range(1, 4):
    d[x] = list()

有没有更好的方法可以做到这一点？假设我直到进入第二个循环才知道我需要的密钥for。例如：

class relation:
    scope_list = list()
...
d = dict()
for relation in relation_list:
    for scope_item in relation.scope_list:
        d[scope_item].append(relation)

那么另一种选择就是更换

d[scope_item].append(relation)

和

if d.has_key(scope_item):
    d[scope_item].append(relation)
else:
    d[scope_item] = [relation,]

处理这个问题的最佳方法是什么？理想情况下，附加会“正常工作”。有没有办法表达我想要一个空列表的字典，即使我第一次创建列表时不知道每个键？

解决方案 1：

您可以使用defaultdict：

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> a = ['1', '2']
>>> for i in a:
...   for j in range(int(i), int(i) + 2):
...     d[j].append(i)
...
>>> d
defaultdict(<type 'list'>, {1: ['1'], 2: ['1', '2'], 3: ['2']})
>>> d.items()
[(1, ['1']), (2, ['1', '2']), (3, ['2'])]

解决方案 2：

您可以使用列表理解来构建它，如下所示：

>>> dict((i, range(int(i), int(i) + 2)) for i in ['1', '2'])
{'1': [1, 2], '2': [2, 3]}

对于问题的第二部分，使用defaultdict

>>> from collections import defaultdict
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> for k, v in s:
        d[k].append(v)

>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]

解决方案 3：

您可以使用setdefault：

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2): 
        d.setdefault(j, []).append(i)

print d  # prints {1: ['1'], 2: ['1', '2'], 3: ['2']}

这个名字相当奇怪的setdefault函数的意思是“获取这个键的值，如果该键不存在，则添加这个值然后返回它。”

正如其他人正确指出的那样，defaultdict是一个更好、更现代的选择。 setdefault在旧版本的 Python（2.5 之前）中仍然有用。

解决方案 4：

就我个人而言，我只是使用 JSON 将内容转换为字符串并转换回来。我理解字符串。

import json
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
mydict = {}
hash = json.dumps(s)
mydict[hash] = "whatever"
print mydict
#{'[["yellow", 1], ["blue", 2], ["yellow", 3], ["blue", 4], ["red", 1]]': 'whatever'}

解决方案 5：

您的问题已经得到解答，但如果我记得清的话，您可以替换如下几行：

if d.has_key(scope_item):

和：

if scope_item in d:

即，在该构造中d引用。有时不是最佳选择（例如，如果您想在与上述关联的之后执行多行代码），我发现语法更易于阅读。d.keys()`defaultdictelseif`in

解决方案 6：

简单的方法是：

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

print(d)
{'1': [1, ], '2':[2, ]}

解决方案 7：

如果从列表创建它，那么您还可以使用带有列表推导的字典推导：

dict_of_list_values = {len(k): [name for name in names if len(name) == len(k)] for k in names}

（我在这个例子中使用了姓名列表）