將滾動 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