在 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]if 仅旋转一列sales。

所以代码是：

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