NumPy / SciPy中的广义累积函数？

Question

在numpy或scipy（或其他库）中是否有一个函数将cumsum和cumprod的概念推广为任意函数。例如，考虑（理论上的）函数

cumf( func, array)

func是一个接受两个浮点数并返回一个浮点数的函数。特殊情况

lambda x,y: x+y

和

lambda x,y: x*y

分别是cumsum和cumprod。例如，如果

func = lambda x,prev_x: x^2*prev_x

我将其应用于：

cumf(func, np.array( 1, 2, 3) )

我想要

np.array( 1, 4, 9*4 )

Answer 1

NumPy的ufunc有accumulate()：

In [22]: np.multiply.accumulate([[1, 2, 3], [4, 5, 6]], axis=1)
Out[22]: 
array([[  1,   2,   6],
       [  4,  20, 120]])

不幸的是，呼吁accumulate()在frompyfunc()“编Python函数失败，一个奇怪的错误：

In [32]: uadd = np.frompyfunc(lambda x, y: x + y, 2, 1)

In [33]: uadd.accumulate([1, 2, 3])
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)

ValueError: could not find a matching type for <lambda> (vectorized).accumulate, 
            requested type has type code 'l'

这是将NumPy 1.6.1与Python 2.7.3结合使用。