我有一个长度为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]
我错过了什么吗?
答案 0 :(得分:-1)
我会使用zip
执行此操作:
a_ = [0] * 2 + a
list(map(sum, zip(a_, a_[1:], a_[2:])))
> [1, 3, 9, 12, 13]