相当于Python中F#的Seq.scan()方法?

时间:2010-05-10 19:04:05

标签: python f# functional-programming

在Python中是否有像F#的Seq.scan()这样的函数?

我想在不循环的情况下做一些cumsum()cumproduct()种事情。

6 个答案:

答案 0 :(得分:5)

不。

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

答案 1 :(得分:5)

我认为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的方式,默认情况下它具有相同的类型。

答案 2 :(得分:2)

如果它只是关于执行cumsum / cumprod操作,那么你应该看看numpy对数组的超高效cumsumcumprod操作。

答案 3 :(得分:0)

汇总函数将使用reduce而不是map

有关详细信息,请参阅http://docs.python.org/library/functions.html

答案 4 :(得分:0)

你可以使用accumulate函数获得与python等效的scan相同的功能:

from itertools import accumulate
nums=[1,2,3,4,5]
nums_added=list(accumulate(nums,lambda x,y:x+y))
print(nums_added)

#prints
# [1,3,6,10,15]

您也可以使用 rx(反应式扩展)来使用扫描运算符。 这是一个外部图书馆。所以你需要在使用它之前安装它。 使用 rx,您可以执行以下操作:

from rx import  from_, operators as op
nums=[1,2,3,4,5]
source=from_(nums).pipe(op.scan(lambda x,y:x+y))
source.subscribe(lambda num:print(num)

#prints
#1
#3
#6
#10
#15

rx 可以做的远不止这些,它允许您进行响应式编程。 要了解更多信息,请访问:http://reactivex.io

答案 5 :(得分:-3)

我不确定但是试一试

map(function, iterable, ...)¶

有关docs.python

的详情