Установка ограничения на количество элементов массива, которые должны быть суммированы?

Question

У меня есть массив длиной 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] 

Я что-то упустил?

Answer 1

Я хотел бы сделать это с помощью zip:

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

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