Is there a function like a F#'s Seq.scan() in Python?
I want to do some cumsum() or cumproduct() kind of things without looping.
解决方案
Ignacio's solution is almost right I think, but requires a operator of type ('a -> 'a -> 'a) and doesn't yield the first element.
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]
Specifically, the type of Seq.scan is
('State -> 'T -> 'State) -> 'State -> seq -> seq
The default approach in Python is to write a scan with the type
('State -> 'State -> 'State) -> seq -> seq
This comes from the way that Python specifies reduce, which has the same type by default.