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

Abh*_*nan 3 time-series series correlation dataframe pandas

我正在尝试使用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)
Run Code Online (Sandbox Code Playgroud)

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

r = s.rolling(5)
Run Code Online (Sandbox Code Playgroud)

获得:

Rolling [window=5,center=False,axis=0]
Run Code Online (Sandbox Code Playgroud)

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

r.corr(other=r)
Run Code Online (Sandbox Code Playgroud)

我只有NaN

我根据文档尝试了另一种方法:

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

得到类似的东西:

...
2018-03-01 a NaN NaN
           b NaN NaN
Run Code Online (Sandbox Code Playgroud)

真的不确定我要怎么做。任何帮助将不胜感激!该文档也使用float64。认为这是因为相关性非常接近零,所以显示NaN吗?有人在这里提出了一个错误报告,但是我认为jreback在以前的错误修复中解决了该问题。

这是另一个相关的答案,但是它使用的是pd.rolling_apply,而Pandas 0.23.3版似乎不支持该功能吗?

raf*_*elc 5

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
Run Code Online (Sandbox Code Playgroud)

  • 有没有更快的方法?当数据帧长 100k 时,这需要很长时间才能运行。谢谢 (2认同)