Python :迭代添加列表的相同元素

Question

我有一个元素列表，如下所示-

[(a,1),(b,2),(c,3),(d,4),(e,5),(f,6),(a,7),(b,8),(c,9),(d,10),(e,11),(f,12)]

我正在尝试将字母旁边给出的值与同一字母旁边给出的不同值相加。

例如，"a">

的值为1，程序应将"a"与列表中的所有术语进行比较，直到找到匹配项。 一旦它找到另一个值为 7 的项 "a"，它应该将这两个值相加，以便我们得到 a=8。

预期输出为-

a=8, b=10, c=12, d=14, e=16, f=18

Answer 1

您建议的解决方案的复杂性为 O(n²(。

您可以使用 defaultdict 在 O(n( 中执行此操作(因为它需要对整个列表进行一次传递(：

from collections import defaultdict
li = [('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e', 5), ('f', 6), ('a', 7), ('b', 8), 
('c', 9), ('d', 10), ('e', 11), ('f', 12)]
output = defaultdict(int)
for letter, number in li:
output[letter] += number
print(output)
# defaultdict(<class 'int'>, {'a': 8, 'b': 10, 'c': 12, 'd': 14, 'e': 16, 'f': 18})

此解决方案当然要求元素是可哈希的，但如果不是，则可以通过使用元素的字符串表示来解决此问题。

Answer 2

不使用默认字典的答案。

_list = [("a",1),("b",2),("c",3),("d",4),("e",5),("f",6),("a",7),("b",8),("c",9),("d",10),("e",11),("f",12)]
tmp = dict()
for k in _list:
try:
tmp[k[0]] += k[1]
except KeyError:
tmp[k[0]] = k[1]
print(tmp)    

结果：

{'a': 8, 'b': 10, 'c': 12, 'd': 14, 'e': 16, 'f': 18}

Answer 3

使用itertools.groupby和operator.itemgetter，可以用lambda代替

from itertools import groupby
from operator import itemgetter
lst = sorted(lst, key=itemgetter(0))
d = {k: sum(i[1] for i in g) for k, g in groupby(lst, key=itemgetter(0))}
# {'a': 8, 'b': 10, 'c': 12, 'd': 14, 'e': 16, 'f': 18}

扩展了字典理解：

d = {}
for k, g in groupby(lst, key=itemgetter(0)):
d[k] = sum(i[1] for i in g)

Answer 4

from collections import defaultdict
l = [('a',1),('b',2),('c',3),('d',4),('e',5),('f',6),('a',7),('b',8),('c',9),('d',10),('e',11),('f',12)]
d = defaultdict(int)
for k, v in l:
d[k] += v

结果：

defaultdict(<class 'int'>, {'a': 8, 'b': 10, 'c': 12, 'd': 14, 'e': 16, 'f': 18})