熊猫groupby和shift但在每个块内

Question

我有一个数据框，然后我将其分组：

ID   week   sale 
A     1     1
A     2     4
A     3     10
B     1     7
B     2     2.3
B     3     4.4

我想申请轮班以取得前一周的销售，但只能停留在同一ID区块中。所需的是：

ID   week   sale   sale-shift
A     1     1      nan
A     2     4      1
A     3     10     4
B     1     7      nan
B     2     2.3    7
B     3     4.4    2.3

如果我使用班次：

df['sale-shift']=df['sale'].shift(-1)

它未考虑ID上的分组依据。我想为每个ID（对于我的客户）执行。

谢谢！

Answer 1

import pandas as pd
from pandas.compat import StringIO

print(pd.__version__)

csvdata = StringIO("""ID,week,sale
A,1,1
A,2,4
A,3,10
B,1,7
B,2,2.3
B,3,4.4""")

df = pd.read_csv(csvdata, sep=",")
df['sale-shift'] = df.groupby('ID')['sale'].shift(-1)
print(df)