首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >Python创建列表字典

Python创建列表字典
EN

Stack Overflow用户
提问于 2009-06-06 22:47:30
回答 7查看 562.9K关注 0票数 236

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

代码语言:javascript
复制
{
  1: ['1'],
  2: ['1','2'],
  3: ['2']
}

如果我这样做了:

代码语言:javascript
复制
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之后添加以下代码来初始化字典。

代码语言:javascript
复制
for x in range(1, 4):
    d[x] = list()

有没有更好的方法来做这件事?假设在进入第二个for循环之前,我不知道需要哪些密钥。例如:

代码语言:javascript
复制
class relation:
    scope_list = list()
...
d = dict()
for relation in relation_list:
    for scope_item in relation.scope_list:
        d[scope_item].append(relation)

一种替代方案就是替换掉

代码语言:javascript
复制
d[scope_item].append(relation)

使用

代码语言:javascript
复制
if d.has_key(scope_item):
    d[scope_item].append(relation)
else:
    d[scope_item] = [relation,]

处理这个问题的最好方法是什么?理想情况下,追加将“恰到好处”。有什么方法可以表达我想要一个空列表的字典,即使我在第一次创建列表时并不知道每个键?

EN

回答 7

Stack Overflow用户

回答已采纳

发布于 2009-06-06 23:00:40

您可以使用defaultdict

代码语言:javascript
复制
>>> 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'])]
票数 305
EN

Stack Overflow用户

发布于 2009-06-06 22:54:34

您可以使用列表理解构建它,如下所示:

代码语言:javascript
复制
>>> dict((i, range(int(i), int(i) + 2)) for i in ['1', '2'])
{'1': [1, 2], '2': [2, 3]}

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

代码语言:javascript
复制
>>> 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])]
票数 59
EN

Stack Overflow用户

发布于 2009-06-06 22:49:50

您可以使用setdefault

代码语言:javascript
复制
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在Python2.5之前的旧版本中仍然很有用。

票数 41
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/960733

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档