按班次分组，保存列

Question

我尝试在移动时保存我groupby所在的列

pd.DataFrame({'column1':['Alex', 'Bob', 'Bob'], 'column2':[3, 4, 5]}).groupby('column1').shift(1)

我希望最终的数据帧在索引中具有'Alex'，'Bob'和'Bob'

Answer 1

您可以将移位后的值添加到现有数据框中：

df = pd.DataFrame({'column1':['Alex', 'Bob', 'Bob'], 'column2':[3, 4, 5]})
df['shift']  = df.groupby(['column1']).shift(1)
df = df.set_index('column1')