python - 重命名 MultiIndex Pandas Dataframe 的名称

Question

我在使用 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)