如何自动移除Pandas多层索引中的冗余层级？

这问题我之前处理多层索引时也碰到过，完全懂你不想手动指定层级、希望自动移除所有值都相同的冗余层级的需求！其实我们可以自己实现一个类似你虚构的drop_redundant()的功能，思路非常直观：遍历MultiIndex的每个层级，检查该层级下的所有值是否唯一，只保留那些存在不同值的层级即可。

实现自定义的drop_redundant函数 #

先写一个可以复用的函数：

import pandas as pd

def drop_redundant_multiindex(index):
    # 筛选出所有值不唯一的层级名称
    keep_levels = [level for level in index.names if index.get_level_values(level).nunique() > 1]
    # 如果所有层级都有不同值，直接返回原索引；否则移除冗余层级
    if len(keep_levels) == len(index.names):
        return index
    return index.droplevel([level for level in index.names if level not in keep_levels])

用你的示例DataFrame测试 #

先构造你给出的示例数据：

df = pd.DataFrame(
    [[1,2],[3,4]], 
    columns=["a", "b"], 
    index=pd.MultiIndex.from_tuples(
        [("a", "b", "c"), ("d", "b", "e")], 
        names=["one", "two", "three"]
    )
)

调用函数处理索引：

df.index = drop_redundant_multiindex(df.index)
print(df)

输出结果完全符合你的预期：

a  b
one three
a   c      1  2
d   e      3  4

额外说明 #

这个函数还能处理多个冗余层级的情况，比如如果你的MultiIndex里有两个层级的所有值都相同，它会自动把这两个层级一起移除，不用做任何额外调整。

内容的提问来源于stack exchange，提问作者Fredrik