Есть ли функция как F# Seq.scan()
в Python?
Я хочу сделать некоторых cumsum()
или cumproduct()
вид вещей без цикличного выполнения.
Решение Игнасио почти правильное, но требует оператора типа ('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
, который по умолчанию имеет тот же тип.
Если речь идет только о выполнении операций cumsum/cumprod, то вам стоит обратить внимание на суперэффективные cumsum и cumprod операции numpy над массивами.
Неа.
def scan(op, seq):
it = iter(seq)
result = next(it)
for val in it:
result = op(result, val)
yield result
Агрегированные функции будут использовать reduce
, а не map
.
См. http://docs.python.org/library/functions.html для получения дополнительной информации