Групповая смена, сохранение столбца

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