使用 Pandas.rolling 计算滚动自相关

2024-03-14

我正在尝试使用 Pandas (0.23.3) 计算 Series 对象的滚动自相关

设置示例:

dt_index = pd.date_range('2018-01-01','2018-02-01', freq = 'B')
data = np.random.rand(len(dt_index))
s = pd.Series(data, index = dt_index)

创建窗口大小 = 5 的滚动对象:

r = s.rolling(5)

Getting:

Rolling [window=5,center=False,axis=0]

现在,当我尝试计算相关性时(很确定这是错误的方法):

r.corr(other=r)

我只得到 NaN

我尝试了另一种基于文档: https://pandas.pydata.org/pandas-docs/stable/computation.html#computing-rolling-pairwise-covariances-and-correlations:

df = pd.DataFrame()
df['a'] = s
df['b'] = s.shift(-1)
df.rolling(window=5).corr()

得到类似的东西:

...
2018-03-01 a NaN NaN
           b NaN NaN

真的不知道我哪里出了问题。任何帮助将不胜感激!该文档也使用 float64。认为这是因为相关性非常接近于零,所以显示为 NaN?有人提出了错误报告here https://github.com/pandas-dev/pandas/issues/18430,但我认为 jreback 在之前的错误修复中解决了该问题。

这是另一个相关答案,但它正在使用pd.rolling_apply https://pandas.pydata.org/pandas-docs/version/0.17.0/generated/pandas.rolling_apply.html,Pandas 0.23.3 版本似乎不支持?


IIUC,

>>> s.rolling(5).apply(lambda x: x.autocorr(), raw=False)

2018-01-01         NaN
2018-01-02         NaN
2018-01-03         NaN
2018-01-04         NaN
2018-01-05   -0.502455
2018-01-08   -0.072132
2018-01-09   -0.216756
2018-01-10   -0.090358
2018-01-11   -0.928272
2018-01-12   -0.754725
2018-01-15   -0.822256
2018-01-16   -0.941788
2018-01-17   -0.765803
2018-01-18   -0.680472
2018-01-19   -0.902443
2018-01-22   -0.796185
2018-01-23   -0.691141
2018-01-24   -0.427208
2018-01-25    0.176668
2018-01-26    0.016166
2018-01-29   -0.876047
2018-01-30   -0.905765
2018-01-31   -0.859755
2018-02-01   -0.795077
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用 Pandas.rolling 计算滚动自相关 的相关文章

随机推荐