python cumsum with reduce函数

Question

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

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不知道(我的累加器参数)是一个列表对象，而我希望我的reduce函数返回一个列表。在其他函数式编程语言中，它可能会要求我指定a和x的类型。但我是python的新手，还没有完全弄清楚它是如何处理类型和内容的。解决这个问题的pythonic方法是什么？

Answer 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 a编写if len(a)>0也更好)