Groupbyshift，保存列

Question

在移位时，我尝试保存Igroupby所在的列

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')