蟒蛇cumsum与减少功能

Z时代
2024-01-10
分类：问答

我想在python中使用reduce函数编写一个累积和的版本。这是我到目前为止的代码：蟒蛇cumsum与减少功能

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

回答：

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

结果：