python - 是否等效于Python中F#的Seq.scan()方法？

Python中是否有类似F#的Seq.scan()的函数？

我想做一些cumsum()或cumproduct()类的事情而无需循环。

最佳答案

我认为Ignacio的解决方案几乎是正确的，但是需要类型('a->'a->'a)的运算符，并且不会产生第一个元素。

def scan(f, state, it):
  for x in it:
    state = f(state, x)
    yield state
# test
>>> snoc = lambda xs,x: xs+[x]
>>> list(scan(snoc, [], 'abcd'))
[['a'], ['a', 'b'], ['a', 'b', 'c'], ['a', 'b', 'c', 'd']]
>>> list(scan(operator.add, 0, [1,2,3]))
[1,3,6]

具体来说，Seq.scan的类型为

('State -> 'T -> 'State) -> 'State -> seq<'T> -> seq<'State>

Python中的默认方法是编写一个类型为scan的代码

('State -> 'State -> 'State) -> seq<'State> -> seq<'State>

这来自Python指定reduce的方式，默认情况下它具有相同的类型。

关于python - 是否等效于Python中F#的Seq.scan()方法？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/2805624/