我正在尝试将功能应用于滚动窗口。我需要从数据框中引用多个列以计算函数的输出。以下是我要执行的操作的一些示例代码。
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)
谢谢
目前rolling
,每个列分别实现,因此不可能将2个不同的列相加。
您可以通过以下方式进行测试:
def func(x):
print (x)
return x.sum()
d.rolling(3).apply(func)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句