与每个部分的累积总和有关的这个问题的最佳方法是什么？

Question

我有这些信息：
[0,1,1,1,1,1,0,1,1,1,0,1,1,1,1,0,1,1,1]

并且只需要在零之间的部分连续添加15，结果：
[0,15,30,45,60,75,0,15,30,45,0,15,30,45,60,0,15,30,45]

Answer 1

import numpy
l = np.array([0,1,1,1,1,1,0,1,1,1,0,1,1,1,1,0,1,1,1])
l *= 15
s = np.hstack([np.cumsum(v) for v in np.split(l, np.where(l==0)[0])])
print(s)

np.split 需要一个要拆分的数组和一个可迭代的剪切。 np.where 为您提供所有索引i 其中l[i] == 0。 np.cumsum 对所有这些进行累积总和，然后 np.hstack 将它们重新组合在一起。

Answer 2

伪代码

create an empty list which will be the result
initialize a counter at 0
for each element of the original list
    add 1 to the counter if the element is not 0 else reset the counter to 0
    append to the result list 15 times the counter