计算每列的Pandas DataFrame的自相关性

Question

我想计算Pandas DataFrame列中滞后长度的自相关系数.我的数据片段是:

            RF        PC         C         D        PN        DN         P
year                                                                      
1890       NaN       NaN       NaN       NaN       NaN       NaN       NaN
1891 -0.028470 -0.052632  0.042254  0.081818 -0.045541  0.047619 -0.016974
1892 -0.249084  0.000000  0.027027  0.067227  0.099404  0.045455  0.122337
1893  0.653659  0.000000  0.000000  0.039370 -0.135624  0.043478 -0.142062

沿着这一年,我想计算每列(RF,PC等)的滞后一的自相关.

为了计算自相关,我为每个列提取了两个时间序列,其开始和结束数据相差一年,然后计算相关系数numpy.corrcoef.

例如,我写道:

numpy.corrcoef(data[['C']][1:-1],data[['C']][2:])

(调用整个DataFrame data).
但是,该命令很遗憾地返回:

array([[ nan,  nan,  nan, ...,  nan,  nan,  nan],
       [ nan,  nan,  nan, ...,  nan,  nan,  nan],
       [ nan,  nan,  nan, ...,  nan,  nan,  nan],
       ..., 
       [ nan,  nan,  nan, ...,  nan,  nan,  nan],
       [ nan,  nan,  nan, ...,  nan,  nan,  nan],
       [ nan,  nan,  nan, ...,  nan,  nan,  nan]])

有人可以告诉我如何计算自相关吗？

Answer 1

这是一个迟到的答案,但对于未来的用户,您还可以使用pandas.Series.autocorr(),它计算Series上的lag-N(默认值= 1)自相关:

df['C'].autocorr(lag=1)

http://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.autocorr.html#pandas.Series.autocorr

Answer 2

.autocorr 适用于Series,而不是DataFrames.您可以使用.apply以应用于DataFrame:

def df_autocorr(df, lag=1, axis=0):
    """Compute full-sample column-wise autocorrelation for a DataFrame."""
    return df.apply(lambda col: col.autocorr(lag), axis=axis)
d1 = DataFrame(np.random.randn(100, 6))

df_autocorr(d1)
Out[32]: 
0    0.141
1   -0.028
2   -0.031
3    0.114
4   -0.121
5    0.060
dtype: float64

您还可以使用指定窗口计算滚动自相关,如下所示(这是.autocorr在幕后进行的操作):

def df_rolling_autocorr(df, window, lag=1):
    """Compute rolling column-wise autocorrelation for a DataFrame."""

    return (df.rolling(window=window)
        .corr(df.shift(lag))) # could .dropna() here

df_rolling_autocorr(d1, window=21).dropna().head()
Out[38]: 
        0      1      2      3      4      5
21 -0.173 -0.367  0.142 -0.044 -0.080  0.012
22  0.015 -0.341  0.250 -0.036  0.023 -0.012
23  0.038 -0.329  0.279 -0.026  0.075 -0.121
24 -0.025 -0.361  0.319  0.117  0.031 -0.120
25  0.119 -0.320  0.181 -0.011  0.038 -0.111

Answer 3

你应该使用：

numpy.corrcoef(df['C'][1:-1], df['C'][2:])

df[['C']]表示只有一列的数据框，而df['C']是包含 C 列中的值的系列。