压缩类似键以构建字典时如何对值求和

Question

我有A = [a, b, c, d, a, d, c]和B=[1, 2, 3, 4, 5, 6, 7]

为什么dict(zip(A,B))不返回{'a': 6, 'b': 2, 'c': 10, 'd': 10} ？

如何使它起作用？

Answer 1

使用简单的迭代

前任：

A = ["a", "b", "c", "d", "a", "d", "c"] 
B= [1, 2, 3, 4, 5, 6, 7]

result = {}
for a, b in zip(A, B):
    if a not in result:
        result[a] = 0
    result[a] += b
print(result)

---

或者使用collections.defaultdict

前任：

from collections import defaultdict
result = defaultdict(int)
for a, b in zip(A, B):
    result[a] += b
pprint(result)

输出：

{'a': 6, 'b': 2, 'c': 10, 'd': 10}

Answer 2

dict只会覆盖值.. 你想要的不会那么容易。 你需要这样的东西：

#!/usr/bin/env python3
from collections import defaultdict

A = ["a", "b", "c", "d", "a", "d", "c"]
B = [1, 2, 3, 4, 5, 6, 7]

output = defaultdict(int)

for a,b in zip(A,B):
        output[a] += b

print(output)

结果：

defaultdict(<class 'int'>, {'a': 6, 'b': 2, 'c': 10, 'd': 10})

默认情况下， defaultdict会将每个新键的值设置为0 .. 允许我们在每个键上调用+=没有错误.. 给我们我们需要的总和。