在 Python pandas 中展平 MultiIndex pivot 表

Question

這是我的 pivot 表列結構（多索引）：

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

Answer 2

我會做這樣的事情，其中“pivot”是您的 pivot 表的名稱：

pivot_flat = pd.DataFrame(pivot.to_records())

這將使您的 pivot 表變平，以便您可以執行更強大的可視化或添加其他計算列和轉換。 從本質上講，你將從這個 go ：

對此：