如何在 python 中聚合每周数据

Question

我有以下格式的全年数据列表：

01 5170
01 3490
02 3630
02 3170

首先我们有周数（01, 02），然后是相应的花费（第 1 周的 5170 和 3490 等）

我将如何 go 关于添加花费的钱，以便我得到在第 1 周、第 2 周等花费的所有钱。在第一周花费的所有钱应该加在一起等等。

所以上面的数据会变成：

01  8660
02  6800

有任何想法吗？

Answer 1

鉴于：

text = """01 5170
01 3490
02 3630
02 3170"""

xss = [[int(x) for x in s.split(" ")] for s in text.splitlines()]

>>> xss
[[1, 5170], [1, 3490], [2, 3630], [2, 3170]]

我们可以排序、分组，然后归约（求和）：

from itertools import groupby

groups = groupby(xss, key=lambda x: x[0])
result = [(k, sum(x for _, x in xs)) for k, xs in groups]

>>> result
[(1, 8660), (2, 6800)]