用python求和多维列表

Question

如何计算：

[["toto", 3], ["titi", 10], ["toto", 2]]

要得到这个：

^{pr2}$

谢谢

Answer 1

您可以使用^{}

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for i, j in L:
...     d[i].append(j)
... 
>>> [[i, sum(j)] for i, j in d.items()]
[['titi', 10], ['toto', 5]]

---

感谢@raymonad提供了另一种更清洁的解决方案：

^{pr2}$

Answer 2

l1=[["toto", 3], ["titi", 10], ["toto", 2]]
d={}
for i in range(len(l1)):
    try:
        d[l1[i][0]]+=l1[i][1]
    except KeyError:
        d[l1[i][0]]=l1[i][1]
l2=[]
for k,v in d.items():
    l2.append([k,v])

Answer 3

您可以使用^{}对第一项进行分组，然后计算总和：

In [1]: data = [["toto", 3], ["titi", 10], ["toto", 2]]

In [2]: from itertools import groupby

In [3]: from operator import itemgetter

In [4]: key = itemgetter(0)

In [5]: [[k, sum(l[1] for l in g)] 
    ..:    for k, g in groupby(sorted(data, key=key), key=key)]
Out[5]: [['titi', 10], ['toto', 5]]