Python:编辑分隔符之间的嵌套列表中的数字总和

Question

我有嵌套列表，看起来像这样 [[1, 0, 0, 1, 1], [1, 1, 1, 0, 1]]。我如何对分隔符(在本例中为零)之间的数字(在本例中为 1)求和以获得如下输出:[[1,2], [3,1]]？

编辑那么列表以 0 开头并且有更多组的情况呢？例如 [0, 1, 0, 1, 1, 0, 1, 1] --> [1, 2, 2]

Answer 1

你可以使用 groupby :

from itertools import groupby

data = [[1, 0, 0, 1, 1], [1, 1, 1, 0, 1]]

separator = 0
result = [[sum(v) for k, v in groupby(element, key=lambda x: x != separator) if k] for element in data]
print(result)

输出

[[1, 2], [3, 1]]