python - 如何在numpy中应用熊猫滚动？

Question

我想将pandas.rolling(window).apply(func) 范式应用于 numpy 数组。在搜索came across this function for numpy 时。这就是我对它所做的。

def rolling_apply(fun, a, w):
    r = np.empty(a.shape)
    r.fill(np.nan)
    for i in range(w - 1, a.shape[0]):
        r[i] = fun(a[(i-w+1):i+1])
    return r

def test(x):
    return x.sum()

arr=np.random.rand(20)
e=rolling_apply(test,arr,10)

运行时出现此错误

ValueError: 使用序列设置数组元素。

你能告诉我为什么会发生这个错误

编辑：

这项工作我在上面的代码中犯了一个初始错误。这是有效的

Answer 1

数组形状和索引有点混乱。快速修复：

def rolling_apply(fun, a, w):
r = np.empty((a.shape[0]-w+1, w))
r.fill(np.nan)
for i in range(w - 1, a.shape[0]):
    r[i-w+1] = fun(a[(i-w+1):i+1])
return r

def test(x):
    return x*2

arr=np.random.rand(20)
e=rolling_apply(test,arr,10)