python - Pandas - 时间序列按组移动

Question

我有一个这样的数据框:

pd.DataFrame({'group': {0: 1, 1: 1, 2: 1, 3: 1, 4: 2, 5: 2, 6: 2}, 'year': {0: 2007, 1: 2008, 2: 2009, 3: 2010, 4: 2006, 5: 2007, 6: 2008}, 'amount': {0: 2.0, 1: 4.0, 2: np.nan, 3: 7.0, 4: 8.0, 5: 10.0, 6: 12.0}}])

   group    year    amount
0   1       2007    2
1   1       2008    4
2   1       2009    NaN
3   1       2010    7
4   2       2006    8
5   2       2007    10
6   2       2008    12

我想为每个组添加最小值(删除 NA)、最大值(删除 NA)、滞后 1 和滞后 2。我理想的数据框是这样的

   group    year    amount    min    max   lag1    lag2
0   1       2007    2          2      7     NaN     NaN
1   1       2008    4          2      7     2       NaN
2   1       2009    NaN        2      7     4       2
3   1       2010    7          2      7     NaN     4
4   2       2006    8          8      12    NaN     NaN
5   2       2007    10         8      12    8       NaN
6   2       2008    12         8      12    10      8

我知道在 panda 中使用 agg 求最小值和最大值，但我不确定如何按组处理滞后

Answer 1

国际联合会

g = df.groupby('group').amount
df['max'] = g.transform('max')
df['min'] = g.transform('min')
df['lag1'] = g.shift()
df['lag2'] = g.shift(2)

group   year    amount  max     min lag1    lag2
0   1   2007    2.0     7.0     2.0 NaN     NaN
1   1   2008    4.0     7.0     2.0 2.0     NaN
2   1   2009    NaN     7.0     2.0 4.0     2.0
3   1   2010    7.0     7.0     2.0 NaN     4.0
4   2   2006    8.0     12.0    8.0 NaN     NaN
5   2   2007    10.0    12.0    8.0 8.0     NaN
6   2   2008    12.0    12.0    8.0 10.0    8.0