过滤pandas.DataFrame中的索引级别

Question

如果我有一个多索引数据帧：

import pandas as pd
df = pd.DataFrame([[1,2,3],[4,5,6],[7,8,9]],columns=['a','b','c']).set_index(['a','b'])

我可以简单地过滤列上的数据帧，例如：

df[df.c>4]

但是要在索引级别上做同样的事情，比如说“b”，我做不到：

df[df.b>4]

相反，我可以这样做：

df[df.index.get_level_values('b')>4]

但是，有一个不那么冗长的方法吗？

Answer 1

您可以使用query以获得更好的可读性

In [795]: df.query('b > 4')
Out[795]:
     c
a b
4 5  6
7 8  9