过滤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