带有reduce函数的python cumsum

【问题标题】：python cumsum with reduce function带有reduce函数的python cumsum
【发布时间】：2017-12-18 15:59:49
【问题描述】：

我正在尝试使用 reduce 函数在 python 中编写累积和的版本。到目前为止，这是我的代码：

from functools import reduce

def my_cum_sum(arg):
    return reduce(lambda a, x: (a.append(a[-1] + x)) if len(a) > 0 else a.append(x), arg, [])

assert(my_cum_sum([1, 1, 1, 1]) == [1, 2, 3, 4]))

但问题是在我的 lambda 函数中，python 不知道 a（我的累加器参数）是一个列表对象，并且我希望我的 reduce 函数返回一个列表。在其他函数式编程语言中，它可能会要求我指定 a 和 x 的类型。但我是 python 新手，还没有完全弄清楚它是如何处理类型和东西的。解决这个问题的pythonic方法是什么？

【问题讨论】：

标签： python functional-programming

【解决方案1】：

append 返回None，因此您无法像这样将列表返回到reduce。只需在列表和由单个元素组成的列表之间使用加法，或者如果列表为空，则仅使用由初始元素组成的列表：

from functools import reduce

def my_cum_sum(arg):
    return reduce(lambda a, x: a + [a[-1]+x] if a else [x], arg, [])

print (my_cum_sum([1, 1, 1, 1]))

结果：

[1, 2, 3, 4]

（注意if len(a)>0 最好写成if a）

【讨论】：