Объединяя карту и уменьшить

Question

я хочу найти изящный способ достичь следующего:

Притворитесь У меня есть список:

> x = [1, 2, 3, 4, 5] 

И простую функцию, которая просто добавляет два числа:

> def add(a, b) 
     return a+b 

Я могу сократить список x непосредственно:

> sum = reduce(add, x) 
> print(sum) 
15 

Который дает мне сумму просто отлично. Но я хотел бы знать значение после каждое приложение добавления. Таким образом, используя функцию, подобную уменьшить, я хотел бы получить обратно следующий массив:

> result = SOME_FUNCTION(add, x) 
> print(result) 
[3, 6, 10, 15] 

ли кто-нибудь есть классный способ достичь этого. У меня есть сильное предпочтение для использования той или иной форме itertools решения, если возможно :)

Answer 1

Так как вы хотите itertools

from itertools import accumulate 
list(accumulate(x)) 
Out [130]: 
[1, 3, 6, 10, 15] 

В качестве альтернативы петлевом Генератор

def cumsum(x): 
    total = 0 
    for x in it: 
     total += x 
     yield total 
list(cumsum(x)) 
Out [129]: 
[1, 3, 6, 10, 15] 

Или просто как уже упоминалось Дэвид:

np.cumsum(x) 
Out [123]: 
array([ 1, 3, 6, 10, 15], dtype=int32)