Сведение сводной таблицы MultiIndex в Python pandas

Question

Вот моя структура столбцов сводной таблицы (мультииндекс):

      col2  col3  col4  sales

month                   month_1  month_2  month_3

Я хотел бы сгладить ее до:

      col2  col3  col4  month_1  month_2  month_3

Если я сделаю pivot.columns = pivot.columns.get_level_values(0), то результат будет :

      col2  col3  col4  sales  sales  sales

Что мне делать?

Answer 1

Я думаю, что решением является удаление [] около [sales] и [months] при повороте только по одному столбцу sales.

Итак, код:

 pivot = (pd.pivot_table(df, 
                         index=['col2','col3','col4'],
                         columns='month', 
                         values='sales')
            .reset_index()
            .rename_axis(None, axis=1))