Можно ли сохранить поля идентификатора при повторной выборке панд datetimeindex?

Question

У меня есть набор данных с индексом datetime и двумя столбцами, ['WorkoutID', 'power'].

Индекс datetime находится на дневном уровне, но я пытаюсь пересчитать до недельного уровня и получить максимальную мощность от всех тренировок в течение недели:

df['best_power'] = df['power'].resample('D').max()

Прямо сейчас вывод этого дает мне datetimeindex на недельном уровне и максимальную мощность ('best_power'), но он отбрасывает столбец workoutID.

Я также хочу видеть, из какого workoutId исходит максимальная мощность - как я могу сохранить столбец workoutID?

Answer 1

Вы можете использовать idxmax с groupby

df.loc[df.groupby(df.index.strftime('%Y-%w')).power.idxmax()]