[英]Python creating a dictionary of lists
我想创建一个值为列表的字典。 例如:
{
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,]
处理这个问题的最佳方法是什么? 理想情况下,附加将“正常工作”。 有什么方法可以表达我想要一个空列表的字典,即使我在第一次创建列表时不知道每个键?
您可以使用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'])]
您可以使用列表理解来构建它,如下所示:
>>> 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])]
您可以使用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 之前)中仍然有用。
就个人而言,我只是使用 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'}
简单的方法是:
a = [1,2]
d = {}
for i in a:
d[i]=[i, ]
print(d)
{'1': [1, ], '2':[2, ]}
您的问题已经得到解答,但 IIRC 您可以替换如下行:
if d.has_key(scope_item):
和:
if scope_item in d:
也就是说, d
在该构造中引用了d.keys()
。 有时defaultdict
不是最佳选择(例如,如果您想在与上述if
关联的else
之后执行多行代码),我发现in
语法更易于阅读。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.