结合地图和减少

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)