具有滚动窗口的Pandas Dataframe枢轴

Question

我正在尝试使用Python Pandas (first timer)为一些时间序列建模准备数据。我的DataFrame看起来像这样：

df = pd.DataFrame({
    'time': [0, 1, 2, 3, 4], 
    'colA': ['a', 'b', 'c', 'd', 'e'], 
    'colB': ['v', 'w', 'x', 'y', 'z'], 
    'value' : [10, 11, 12, 13, 14]
})

#       time  colA  colB  value
# 0     0     a     v     10
# 1     1     b     w     11
# 2     2     c     x     12
# 3     3     d     y     13
# 4     4     e     z     14

有没有可以将其转换为以下格式的函数组合？

# colA-2  colA-1  colA    colB-2  colB-1  colB  value
# _       _       a       _       _       v     10
# _       a       b       _       v       w     11
# a       b       c       v       w       x     12
# b       c       d       w       x       y     13
# c       d       e       x       y       z     14

我对Python/Pandas非常陌生，我没有任何具体的代码/结果让我接近我需要的东西……

Answer 1

您可以使用shift函数：

df['colA-2'] =df['colA'].shift(2, fill_value='-' )
df['colA-1'] =df['colA'].shift(1,fill_value='-')
... 

Answer 2

我会使用pd.concat

pd.concat([
    df[['colA', 'colB']].shift(i).add_suffix(f'-{i}')
    for i in range(1, 3)], axis=1
).fillna('-').join(df)

  colA-1 colB-1 colA-2 colB-2  time colA colB  value
0      -      -      -      -     0    a    v     10
1      a      v      -      -     1    b    w     11
2      b      w      a      v     2    c    x     12
3      c      x      b      w     3    d    y     13
4      d      y      c      x     4    e    z     14