Python列表分组键

Question

我有一个清单：

x = [[17, 2], [18, 4], [17, 2], [18, 0], [19, 4],
     [17, 4], [19, 4], [17, 4], [20, 4], [17, 4],
     [20, 4], [17, 4], [17, 4], [18, 4], [17, 4]]

当第一个值相同时，我想对所有第二个值求和。

例如：17 = 28...

我尝试用 :

d = {}
for row in x:
    if row[0] not in d:
        d[row[0]] = []
    d[row[0]].append(row[1])

结果是

{17: [2, 2, 4, 4, 4, 4, 4, 4],
 18: [4, 0, 4], 19: [4, 4],
 20: [4, 4]}

我没有找到对这些值求和的方法。

Answer 1

如果列表已排序，您可以使用itertools.groupby（并且您可以使用sorted 来确保）：

from itertools import groupby
from operator import itemgetter

d = {key: sum(grp) for key, grp in groupby(sorted(x, key=itemgetter(0)))}

在这种情况下，itemgetter(0) 是lambda x: x[0] 的更有效的快捷方式。

在您原来的情况下，您可以保持一个运行总和或之后的总和。总结你已经拥有的字典：

d = {k: sum(v) for k, v in d.items()}

要保持流动总和：

d = {}
for k, v in x:
    if k in d:
        d[k] += v
    else:
        d[k] = v

做同样事情的更短的方法是使用dict.setdefault：

d = {}
for k, v in x:
    d[k] = d.setdefault(k, 0) + v

Answer 2

使用默认字典：

>>> from collections import defaultdict
>>> d = defaultdict(int)
>>> for lst in x:
...   a, b = lst
...   d[a] +=  b
>>> d
defaultdict(int, {17: 28, 18: 8, 19: 8, 20: 8})

Answer 3

我们在这里：

x=[[17, 2], [18, 4], [17, 2], [18, 0], [19, 4], [17, 4], [19, 4], [17, 4], [20, 4], [17, 4], [20, 4], [17, 4], [17, 4], [18, 4], [17, 4]]
d = {}
for row in x:
    if row[0] not in d:
        d[row[0]] = 0
    d[row[0]] += row[1]

print(d)

输出是

{17: 28, 18: 8, 19: 8, 20: 8}