将滚动 window 添加到 pandas 中每一行的列

Question

我在 dataframe 中有一个时间序列，并且想将带有 window 大小 n 的滚动 window 添加到每一行。

df['A'].rolling(window=6)

这意味着每行将有额外的 6 列，该列的滚动 window 的相应编号。 如何使用 pandas 的滚动 function 自动命名列 [t-1, t-2, t-3...tn] 来实现这一点？

Answer 1

df.shift(n)是您要查找的内容：

n = 6 # max t-n
for t in range(1, n+1):
    df[f'A-{t}'] = df['A'].shift(t)

参考。 https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.shift.html