设置要累积求和的数组元素数量的限制？

Question

我有一个长度为5的数组。

array = [1,2,6,4,3]

我想要的是一个新的数组，累计地求和以前值的最大集合数，即3：

return_array = [1,3,9,12,13]

使用numpy的“累积和”函数似乎是一个很好的起点，但它似乎仅限于对所有元素进行求和，给出如下结果：

cumsum_array = np.cumsum(array)
cumsum_array = [1,3,9,13,16]

我是不是遗漏了什么？

Answer 1

我会用zip来做这个

a_ = [0] * 2 + a
list(map(sum, zip(a_, a_[1:], a_[2:])))

> [1, 3, 9, 12, 13]