大熊貓的前一組均值漂移

Question

我有一張看起來像這樣的表：

date         revenue
2021-01-01   20
2021-01-02   30
...
2021-01-31   50
2021-02-01   35
2021-02-02   67

我想計算上個月每一行的最大收入。

我可以計算當月的最大收入：

df['max']=df.groupby(df['date'].dt.month)[revenue'].transform(max)

它看起來像這樣：

  date         revenue  max
2021-01-01   20         50
2021-01-02   30         50
...
2021-01-31   50         50
2021-02-01   35         67
2021-02-02   67         67

但我希望它是：

  date         revenue  max
2021-01-01   20         nan
2021-01-02   30         nan
...
2021-01-31   50         nan
2021-02-01   35         50
2021-02-02   67         50

我試圖把.shift()放在最后，但它只會按行滯后/領先，但我需要按組。

請幫忙

Answer 1

您可以使用shift map進行groupby

s = df.date.dt.strftime('%Y%m')
df['new'] = s.map(df.groupby(s).revenue.max().shift())
df
Out[62]: 
        date  revenue   new
0 2021-01-01       20   NaN
1 2021-01-02       30   NaN
2 2021-01-31       50   NaN
3 2021-02-01       35  50.0
4 2021-02-02       67  50.0