結合地圖和減少

Question

我想找到一個巧妙的方法來實現以下目標：

假裝我有一個列表：

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

和一個簡單的功能，只是增加了兩個數字：

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

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

：

我可以直接減少列表

這給我的總和就好了。但是我想知道每次申請後的價值加。因此，使用類似於reduce的函數，我想返回以下數組：

> 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)