在 Python 的 Pandas 中，如何在 MultiIndex 中的字段上过滤 DataFrame

Question

我有这个 DataFrame 有一个 MultiIndex 字段为Hi ， Md ， Lo

import pandas as pd

df = pd.DataFrame(
    [
        ["A", "X", "C", "7", "3"],
        ["A", "B", "C", "7", "2"],
        ["A", "B", "C", "7", "1"],
        ["A", "X", "C", "7", "3"],
        ["1", "2", "3", "7", "3"],
        ["4", "5", "6", "7", "3"],
    ],
    columns=["Hi", "Md", "Lo", "O", "I"],
).set_index(["Hi", "Md", "Lo"])

看起来像这样

          O  I
Hi Md Lo      
A  X  C   7  3
   B  C   7  2
      C   7  1
   X  C   7  3
1  2  3   7  3
4  5  6   7  3

如何设置过滤器以仅获取Md值为B的行

我尝试通过字段名称

df[df['Md'] == 'B']

这会引发 KeyError，因为“字段”不是列，而是 MultiIndex 的一部分

df[df.index.map(lambda x: x[1] == 'B')]

有效，但非常hacky。

是否有更短的方法来编写此过滤器，可能使用字段名称Md或索引 position 1 ？

Answer 1

您可以使用DataFrame.query

df.query("Md == 'B'")

Output

          O  I
Hi Md Lo      
A  B  C   7  2
      C   7  1

替代方案（可能不适用于不同的数据）

df.filter(like='B', axis=0)

Answer 2

这是一种将元组传递给df.loc的方法

df.loc[(slice(None),'B',slice(None))]