在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 )
https://stackoverflow.com/questions/13828599
复制相似问题