+-

我有一个元组列表:
[ ('A',100), ('B',50), ('A',50), ('B',20), ('C',10) ]
我试图总结所有具有相同字母的数字.即我想输出
[('A', 150), ('B', 70), ('C',10)]
我尝试使用set获得唯一值,但是当我尝试将第一个元素与set比较时,我得到
TypeError: unsupported operand type(s) for +: 'int' and 'str'
有什么快速解决方案可以按字母匹配数字?
最佳答案
这是一个(一个半?)划线:按字母分组(您需要对其进行排序),然后取元组的第二个条目的总和.
from itertools import groupby
from operator import itemgetter
data = [('A', 100), ('B', 50), ('A', 50), ('B', 20), ('C', 10)]
res = [(k, sum(map(itemgetter(1), g)))
for k, g in groupby(sorted(data, key=itemgetter(0)), key=itemgetter(0))]
print(res)
// => [('A', 150), ('B', 70), ('C', 10)]
以上是O(n log n)-排序是最昂贵的操作.如果您的输入列表确实很大,那么以下O(n)方法可能会更好地为您服务:
from collections import defaultdict
data = [('A', 100), ('B', 50), ('A', 50), ('B', 20), ('C', 10)]
d = defaultdict(int)
for letter, value in data:
d[letter] += value
res = list(d.items())
print(res)
// => [('B', 70), ('C', 10), ('A', 150)]
点击查看更多相关文章
转载注明原文:python-元组列表中按字母求和的数字 - 乐贴网