python - 如果元组具有相同的第一个元素，如何聚合元组列表的元素？

Question

我有一个列表，其中每个值都是一个元组列表。例如，这是我为一个键提取的值:

     [('1998-01-20',8) , ('1998-01-22',4) , ('1998-06-18',8 ) , ('1999-07-15' , 7), ('1999-07-21',1) ]

我也对列表进行了排序。现在我想像这样聚合值:

    [('1998-01' , 12 ) , ('1998-06' ,8 ) , ('1999-07',8 )]

在某种意义上，我想按月对元组进行分组，将每个月的整数加在一起，我已经阅读了有关 groupby 的内容，但我认为它对我的数据结构没有帮助，因为我不知道是什么我将在我的列表中面对，所以我试图找到一种方法来表达:如果 i[0][:6] 相等，则从元组的第一个元素开始:对 i[1] 求和。但我很难实现这个想法。

    for i in List :
        if i[0][:6] # *problem* I don't know how to say my condition :
        s=sum(i[1]) #?

我将不胜感激任何建议，因为我是 python 的新用户!

Answer 1

尝试使用 itertools.groupby 按月汇总值:

from itertools import groupby
a = [('1998-01-20', 8), ('1998-01-22', 4), ('1998-06-18', 8), 
     ('1999-07-15', 7), ('1999-07-21', 1)]

for key, group in groupby(a, key=lambda x: x[0][:7]):
    print key, sum(j for i, j in group)

# Output

1998-01 12
1998-06 8
1999-07 8

这是单行版本:

print [(key, sum(j for i, j in group)) for key, group in groupby(a, key=lambda x: x[0][:7])]

# Output

[('1998-01', 12), ('1998-06', 8), ('1999-07', 8)]