使用多列的熊猫数据框滚动应用功能

Question

我正在尝试将函数应用于滚动窗口。我需要从数据框中引用多个列来计算函数的输出。 下面是我正在尝试做的一些示例代码。

import pandas as pd
import statsmodels.formula.api as smf

d = pd.DataFrame({
    'Col1': range(0,10),
    'Col2': range(10,20),
    'Col3': range(20,30)
    })

func = lambda x: smf.ols('Col1 ~ Col2', data=x).fit().params[1]

d['Out'] = d.rolling(3).apply(func)

谢谢

Answer 1

当前实现 rolling 分别处理每一列，因此不可能对 2 个不同的列求和。

您可以通过以下方式对其进行测试：

def func(x):
    print (x)
    return x.sum()

d.rolling(3).apply(func)