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 )