如何在Pandas多层索引DataFrame中按索引键获取数据截面？

错误原因分析 #

你出错的核心问题是xs方法的level参数传错了：

• df.index.levels[0]返回的是索引级别对应的取值数组（比如这里是Index(['d', 'e'], dtype='object')）
• 但xs的level参数需要的是索引的名称（比如你的索引列名是'A'）或者级别位置（整数，比如0，因为只有一个索引层级）

把取值数组传给level，会导致pandas无法解析参数，从而抛出ValueError，错误里的a就是你传入的这个多元素数组。

正确实现步骤 #

1. 先确认DataFrame的构建（需将A列设为索引） #

import pandas as pd

# 构建并设置索引
df = pd.DataFrame({
    'A': ['d', 'd', 'e', 'e'],
    'B': [1, 3, 5, 7],
    'C': [2, 4, 6, 8]
}).set_index('A')

2. 单索引键提取数据截面 #

用xs方法时，正确指定level参数：

• 方式一：通过索引名称指定

# 获取索引键'd'对应的B、C数据
df.xs(key='d', level='A')

• 方式二：通过级别位置指定（单层级索引时位置为0）

df.xs(key='d', level=0)

两种方式都会得到你想要的结果：

B  C
A      
d  1  2
d  3  4

3. 遍历所有索引键提取数据 #

你可以遍历df.index.levels[0]或者df.index.unique()（后者更直观），结合正确的xs调用：

# 遍历所有唯一索引键
for key in df.index.unique():
    # 提取对应截面
    section = df.xs(key=key, level='A')
    print(f"索引键{key}的截面数据：")
    print(section)
    print("-"*10)

执行后会依次输出：

索引键d的截面数据：
   B  C
A      
d  1  2
d  3  4
----------
索引键e的截面数据：
   B  C
A      
e  5  6
e  7  8
----------

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