Эквивалентный Seq.scan F# () метод в Python?

Есть ли функция как F# Seq.scan() в Python?

Я хочу сделать некоторых cumsum() или cumproduct() вид вещей без цикличного выполнения.

8
задан Peter O. 26 August 2014 в 04:28
поделиться

5 ответов

Решение Игнасио почти правильное, но требует оператора типа ('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, который по умолчанию имеет тот же тип.

5
ответ дан 5 December 2019 в 18:57
поделиться

Если речь идет только о выполнении операций cumsum/cumprod, то вам стоит обратить внимание на суперэффективные cumsum и cumprod операции numpy над массивами.

2
ответ дан 5 December 2019 в 18:57
поделиться

Неа.

def scan(op, seq):
  it = iter(seq)
  result = next(it)
  for val in it:
    result = op(result, val)
    yield result
5
ответ дан 5 December 2019 в 18:57
поделиться

Я не уверен, но попробуйте

map(function, iterable, ...)¶

Подробнее об этом на docs.python

-3
ответ дан 5 December 2019 в 18:57
поделиться

Агрегированные функции будут использовать reduce, а не map.

См. http://docs.python.org/library/functions.html для получения дополнительной информации

0
ответ дан 5 December 2019 в 18:57
поделиться
Другие вопросы по тегам:

Похожие вопросы: