重命名 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表示在多索引中可以找到要重命名的键的位置。