如何对以key为tup的python字典中的值求和

2024-03-28 14:16:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我有下面的字典,键是元组

D_grpTagReport = { ('Tag1', '1'):345.56 ,
                   ('Tag1', '2'):45.6 ,
                   ('Tag2', '3'):3.1 ,
                   ('Tag3', '1'):21.56 ,
                   ('Tag2', '3'):1.56 ,
                 }

我想得到每个唯一标记的值的总和。是否有任何内置实用程序可用于处理此词典。你知道吗

结果示例:

Tag1 : 391.16     # (total of all Tag1)
Tag2 : 4.66       # (total of all Tag2) 
Tag3 : 21.56      # (total of all Tag3) 

Tags: of标记实用程序示例字典all内置词典
3条回答

可以使用^{},首先将.items()作为字典,然后根据键的第一个元素对列表进行排序,并将^{}作为键的第一个元素(元组key,value的第一个元素)。然后,对于每个组,您可以对值进行sum()运算以获得所需的值。示例-

from itertools import groupby
for key,group in groupby(sorted(D_grpTagReport.items()),key=lambda x:x[0][0]):
    total = sum(g[1] for g in group)
    print(key,':',total)

请注意,您的结果有点错误,您的字典也有点错误,因为dictionary不能有多个值的同一个键。你知道吗

演示-

>>> D_grpTagReport = { ('Tag1', '1'):345.56 ,
...                    ('Tag1', '2'):45.6 ,
...                    ('Tag2', '3'):3.1 ,
...                    ('Tag3', '1'):21.56 ,
...                    ('Tag2', '3'):1.56 ,
...                  }
>>>
>>>
>>> from itertools import groupby
>>> for key,group in groupby(sorted(D_grpTagReport.items()),key=lambda x:x[0][0]):
...     total = sum(g[1] for g in group)
...     print(key,':',total)
...
Tag1 : 391.16
Tag2 : 1.56
Tag3 : 21.56

您可以使用defaultdict

>>> from collections import defaultdict
>>> D_grpTagReport = { ('Tag1', '1'):345.56 ,
...                    ('Tag1', '2'):45.6 ,
...                    ('Tag2', '3'):3.1 ,
...                    ('Tag3', '1'):21.56 ,
...                    ('Tag2', '3'):1.56 ,
...                  }
>>> result = defaultdict(int)
>>> for t in D_grpTagReport:
...     result[t[0]] += D_grpTagReport[t]
...
>>> result
defaultdict(<class 'int'>, {'Tag3': 21.56, 'Tag1': 391.16, 'Tag2': 1.56})

已经有一些很好的答案了,但这里有一个不同的(更简单的IMO)方法。你知道吗

d = {('A', '1'): 2, ('A', '2'): 5, ('B', '1'): 1}
keys={k[0]:0 for k in d.keys()}
for key in d:
    keys[key[0]] = keys[key[0]] + d[key]
print(keys)  # {'A': 7, 'B': 1}

再次使用OP的数据:

d = { ('Tag1', '1'):345.56 ,
      ('Tag1', '2'):45.6 ,
      ('Tag2', '3'):3.1 ,
      ('Tag3', '1'):21.56 ,
      ('Tag2', '3'):1.56 ,
    }
keys={k[0]:0 for k in d.keys()}
for key in d:
    keys[key[0]] = keys[key[0]] + d[key]
print(keys)  # {'Tag1': 391.16, 'Tag2': 1.56, 'Tag3': 21.56}

相关问题 更多 >