重命名 MultiIndex 的名稱 Pandas Dataframe

Question

我遇到了從 function 組創建的groupby的麻煩。

df = base.groupby(['year', 'categ']).agg({'id_prod':'count', 'price':'sum'}).unstack(level=1)

它返回這個結果：

但我想將id_prod和price重命名為no_sales和revenue但我不知道該怎么做，因為 MultiIndex

使用print(df.columns)結果是：

MultiIndex([('id_prod', 0),
            ('id_prod', 1),
            ('id_prod', 2),
            (  'price', 0),
            (  'price', 1),
            (  'price', 2)],
           names=[None, 'categ'])

所以是這個names=[]我想改 謝謝你的幫助！

Answer 1

簡短而簡單，它只會根據多索引重命名您的列

df.columns = df.columns.map('_'.join)

Answer 2

df = df.rename(columns={'id_prod': 'no_sales', 'price': 'revenue'}, level=0)

level=0表示在多索引中可以找到要重命名的鍵的位置。