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

时间:2015-04-09 11:56:24

标签: python arrays numpy cumulative-sum

我有一个长度为5的数组,例如

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

我想要的是一个新数组,累计求和先前值的最大设定数,即3:

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

使用numpy的'cumsum'功能似乎是一个很好的起点,但它似乎仅限于对完整的元素范围求和,给出:

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

我错过了什么吗?

1 个答案:

答案 0 :(得分:-1)

我会使用zip执行此操作:

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

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