pandas滚动值

Question

如何获得pandas系列值的某个长度n的滚动值？

例如，如果我有以下内容：

df = pd.DataFrame({'temperature': [0, 1, 2, np.nan, 4, 2, 0.8, 4, 8.8, 7.12]})

我如何获得长度为n的移动值，即类似，如果n = 3：

[NaN，NaN，0]，[NaN，0，1]，…，[4，8.8，7.12]

编辑：如果我使用熊猫滚动，如：

roll = pd.Series.rolling(df, 3).mean()

然后滚动是该系列的移动平均线。在这里，我不希望每个移动的3个值集合的平均值，而是这些3个值的集合。

Answer 1

我认为您需要先添加NaNs，然后再执行以下解决方案：

N = 3
x = np.concatenate([[np.nan] * (N-1), df['temperature'].values])

def rolling_window(a, window):
    shape = a.shape[:-1] + (a.shape[-1] - window + 1, window)
    strides = a.strides + (a.strides[-1],)
    return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides)
print (rolling_window(x, N))
[[  nan   nan  0.  ]
 [  nan  0.    1.  ]
 [ 0.    1.    2.  ]
 [ 1.    2.     nan]
 [ 2.     nan  4.  ]
 [  nan  4.    2.  ]
 [ 4.    2.    0.8 ]
 [ 2.    0.8   4.  ]
 [ 0.8   4.    8.8 ]
 [ 4.    8.8   7.12]]