Skip to main content
 首页 » 编程设计

Python:如何对列表中的元素进行分组并获取组的总和

2025年05月04日21bjzhanghao

我想按值对列表数字进行分组并计算每组的总和。基本上,它看起来是这样的:

输入:

list = [0,1,1,0,0,0,1,0,1,1,1,1,0] 

OUT(我想得到):

newList = [0,2,0,1,0,4,0] 

有什么想法可以用 Python 实现吗?

谢谢

请您参考如下方法:

你可以使用itertools.groupby:

import itertools 
s = [0,1,1,0,0,0,1,0,1,1,1,1,0] 
final_s = [sum(b) for _, b in itertools.groupby(s)] 

输出:

[0, 2, 0, 1, 0, 4, 0]