Как создать фрейм данных со столбцом, включенным в предложение groupby?

Question

У меня есть фрейм данных. Имеет 3 столбца А, Количество. Я сделал группу, используя «A». Теперь я хочу вставить эти значения в новый фрейм данных. Как мне этого добиться?

top_plt=pd.DataFrame(top_plt.groupby('A')['Amount'].sum())

Результирующий фрейм данных содержит только столбец Amount, но столбец groupby «A» отсутствует. Пример: Результат:

Answer 1

DataFrame конструктор не нужен, лучше добавить as_index=False к groupby:

top_plt= top_plt.groupby('A', as_index=False)['Amount'].sum()

Или добавить DataFrame.reset_index:

top_plt= top_plt.groupby('A')['Amount'].sum().reset_index()