python - 将前 X 行变成附加列 [Pandas]

Question

例如，假设我有一列“a”

我想为每个样本/行提供指定量的先前值的“回溯”。例如，回顾 3 应该执行以下操作...

  a
0  1
1  2
2  3
3  4
4  5

   a  a1  a2  a3
0  1  Nan Nan Nan
1  2  1   Nan Nan
2  3  2   1   Nan
3  4  3   2   1
4  5  4   3   2

Answer 1

使用Series.shift与f-strings:

n = 3
for c in range(1, n+1):
    df[f'a{c}'] = df['a'].shift(c)
print (df)
   a   a1   a2   a3
0  1  NaN  NaN  NaN
1  2  1.0  NaN  NaN
2  3  2.0  1.0  NaN
3  4  3.0  2.0  1.0
4  5  4.0  3.0  2.0

列表理解的替代方案和 concat ，最后添加到原来的DataFrame.join :

n = 3
r = range(1, n+1)
df = df.join(pd.concat([df['a'].shift(c) for c in r], axis=1, keys=r).add_prefix('a'))
print (df)
   a   a1   a2   a3
0  1  NaN  NaN  NaN
1  2  1.0  NaN  NaN
2  3  2.0  1.0  NaN
3  4  3.0  2.0  1.0
4  5  4.0  3.0  2.0