Python如何在二维字典中添加所有值并返回一个带有求和值的字典？

Question

我正在开发的部分程序有一个长度为 n 的2D dict。

字典示例：

test_dict = {
0: {'A': 2, 'B': 1, 'C': 5}, 
1: {'A': 3, 'B': 1, 'C': 2}, 
2: {'A': 1, 'B': 1, 'C': 1}, 
3: {'A': 4, 'B': 2, 'C': 5}
}

所有字典都有相同的键但不同的值。我需要将所有值相加如下。

我尝试使用以下方法合并字典：

new_dict = {}
for k, v in test_dict.items():
    new_dict.setdefault(k, []).append(v) 

我也尝试过使用：

new_dict = {**test_dict[0], **test_dict[1], **test_dict[2], **test_dict[3]}

很遗憾，我没有得到想要的结果。

期望的结果： outcome = {'A': 10, 'B': 5, 'C': 13}

如何将所有值添加到单个字典中？

Answer 1

使用pandas的解决方案

将您的dict 转换为pandas.DataFrame，然后对列进行求和并将其转换回dict。

import pandas as pd

df = pd.DataFrame.from_dict(test_dict, orient='index')
print(df.sum().to_dict())

输出：

{'A': 10, 'B': 5, 'C': 13}

替代解决方案

使用collections.Counter，它允许您在dict 中添加相同键的值

from collections import Counter

d = Counter()    
for _,v in test_dict.items():
    d.update(v)

print(d)