将滚动窗口添加到熊猫每行中的列

尼克皮克

我在一个数据帧中有一个时间序列,想向窗口的每一行添加滚动窗口,窗口大小为n。

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

这意味着每一行将具有另外的6列,并具有该列的滚动窗口的相应编号。如何使用熊猫的滚动功能自动命名列[t-1,t-2,t-3 ... tn],以实现此目的?

格热哥兹·斯基宾斯基

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章