¿Equivalente al método Seq.scan() de F # en Python?

Question

¿Existe una función como la de F # 'Seq.scan() en Python?

Quiero hacer algo de cumsum() o cumproduct() tipo de cosas sin bucles.

Answer 1

solución de Ignacio es casi correcta, creo, pero requiere un operador de tipo ('a ->' a -> 'a) y no produce el primer elemento.

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] 

En concreto, el tipo de Seq.scan es

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

El método por defecto en Python es escribir un scan con el tipo

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

Esto viene de la forma en que Python especifica reduce , que tiene el mismo tipo por defecto.

Answer 2

Agregar funciones usaría reduce en lugar de map.

Ver http://docs.python.org/library/functions.html para obtener más información

Answer 3

No estoy seguro, pero le daría prueba

map(function, iterable, ...)¶ 

Más sobre esto en docs.python

Answer 4

Nop.

def scan(op, seq): 
    it = iter(seq) 
    result = next(it) 
    for val in it: 
    result = op(result, val) 
    yield result 

Answer 5

Si solo se trata de realizar operaciones de cumsum/cumprod, entonces debe observar las operaciones súper eficientes de numpy cumsum y cumprod en las matrices.