python - 使用字典推导式添加和组合值？

Question

假设我有一个列表:

a_list = [["Bob", 2], ["Bill", 1], ["Bob", 2]]

我想将它们添加到字典中并将这些值组合到相应的键中。所以，在这种情况下，我想要一个如下所示的字典:

{"Bob" : 4, "Bill" : 1}

如何使用字典理解来做到这一点？

这就是我所拥有的:

d1 = {group[0]: int(group[1]) for group in a_list}

Answer 1

要使用字典理解完成您想要的操作，您需要一个外部额外字典来跟踪到目前为止每个名称的值:

memory = {}
{name: memory[name] for name, count in a_list if not memory.__setitem__(name, count + memory.setdefault(name, 0))}

但这会产生两个字典，其总和为:

>>> a_list = [["Bob", 2], ["Bill", 1], ["Bob", 2]]
>>> memory = {}
>>> {name: memory[name] for name, count in a_list if not memory.__setitem__(name, count + memory.setdefault(name, 0))}
{'Bob': 4, 'Bill': 1}
>>> memory
{'Bob': 4, 'Bill': 1}

这是因为如果没有内存字典，您就无法访问每个名称的运行总和。

此时您也可以只使用字典和常规循环:

result = {}
for name, count in a_list:
    result[name] = result.get(name, 0) + count

或collections.defaultdict() object :

from collections import defaultdict

result = defaultdict(int)
for name, count in a_list:
    result[name] += count

甚至是collections.Counter() object ，为您提供额外的多集功能供以后使用:

from collections import Counter

result = Counter()
for name, count in a_list:
    result[name] += count

另一个效率较低的选项是先对 a_list 进行排序，然后使用 itertools.groupby)() :

from itertools import groupby
from operator import itemgetter

key = itemgetter(0)  # sort by name
{name: sum(v[1] for v in group)
 for name, group in groupby(sorted(a_list, key=key), key)}

这是一种 O(NlogN) 方法与不带排序的简单循环 O(N) 方法。