在Python迭代期间添加新键或追加到字典中旧键的最有效方法？

Question

这是在不同来源的字典中编译数据时的常见情况：

假设您有一个存储事物列表的字典，例如我喜欢的事物：

likes = {
    'colors': ['blue','red','purple'],
    'foods': ['apples', 'oranges']
}

以及第二个字典，其中包含一些相关值：

favorites = {
    'colors':'yellow',
    'desserts':'ice cream'
}

然后您想要迭代“收藏夹” “对象，并使用“喜欢”字典中的适当键将该对象中的项目附加到列表中，或者向其添加一个新键，其值是包含“收藏夹”中的值的列表。

有几种方法可以做到这一点：

for key in favorites:
    if key in likes:
        likes[key].append(favorites[key])
    else:
        likes[key] = list(favorites[key])

或

for key in favorites:
    try:
        likes[key].append(favorites[key])
    except KeyError:
        likes[key] = list(favorites[key])

还有更多...

我通常使用第一种语法，因为它感觉更Pythonic，但如果有其他更好的方法，我很想知道它们是什么。谢谢！

Answer 1

使用 collections.defaultdict，其中默认值是一个新的 list 实例。

>>> import collections
>>> mydict = collections.defaultdict(list)

通过这种方式，调用 .append(...) 将始终成功，因为如果键不存在，append 将在新的空列表上调用。

您可以使用之前生成的列表实例化 defaultdict，以防您从其他来源获取字典 likes，如下所示：

>>> mydict = collections.defaultdict(list, likes)

请注意，使用 list因为 defaultdict 的 default_factory 属性也在 文档。

Answer 2

使用 collections.defaultdict：

import collections

likes = collections.defaultdict(list)

for key, value in favorites.items():
    likes[key].append(value)

defaultdict 采用单个参数，即一个用于根据需要为未知键创建值的工厂。 list 就是这样一个函数，它创建空列表。

迭代 .items() 将使您免于使用键来获取值。

Answer 3

除了 defaultdict 之外，常规 dict 还提供了一种可能性（可能看起来有点奇怪）： dict.setdefault(k[, d])：

for key, val in favorites.iteritems():
    likes.setdefault(key, []).append(val)

谢谢您的 +20 代表 - 我从 1989 年开始30 秒后到 2009 年。让我们记住，欧洲柏林墙倒塌已有 20 年了。

Answer 4

>>> from collections import defaultdict
>>> d = defaultdict(list, likes)
>>> d
defaultdict(<class 'list'>, {'colors': ['blue', 'red', 'purple'], 'foods': ['apples', 'oranges']})
>>> for i, j in favorites.items():
    d[i].append(j)

>>> d
defaultdict(<class 'list'>, {'desserts': ['ice cream'], 'colors': ['blue', 'red', 'purple', 'yellow'], 'foods': ['apples', 'oranges']})

Answer 5

所有答案都是 defaultdict，但我不确定这是最好的方法。向需要字典的代码提供 defaultdict 可能会很糟糕。 （请参阅：如何使 defaultdict 对于意外情况安全客户？）我个人对这个问题感到左右为难。 （我实际上发现这个问题正在寻找“哪个更好，dict.get()”或“defaultdict”的答案）另一个线程中的某人说你不如果您不希望一直出现这种行为，则需要一个 defaultdict，这可能是正确的。也许为了方便而使用 defaultdict 是错误的方法。我认为这里有两个需求被混为一谈：

“我想要一个默认值为空列表的字典。”其中 defaultdict(list) 是正确的解决方案。

和

“如果存在，我想将其追加到该键的列表中，如果不存在，则创建一个列表。”答案是 my_dict.get('foo', []) 和 append() 。

你们觉得怎么样？