在多级索引的一个级别上过滤pandas数据帧

Question

如果我有一个带有多级索引的pandas数据帧,我该如何按该索引的某个级别进行过滤.例如:

df = pd.DataFrame({"id": [1,2,1,2], "time": [1, 1, 2, 2], "val": [1,2,3,4]})
df.set_index(keys=["id", "time"], inplace=True)

我想做的事情如下:

df[df["time"] > 1]

但time不再是专栏.我可以将它保留为列,但我不想拖动数据副本.

Answer 1

In [17]: df[df.index.get_level_values('time') > 1]
Out[17]:
         val
id time
1  2       3
2  2       4

@ piRSquared的解决方案虽然更具惯用性......

Answer 2

query

df.query('time > 1')

         val
id time     
1  2       3
2  2       4

IndexSlice

DataFrame索引必须是lexsorted

df.sort_index().loc[pd.IndexSlice[:, 2:], :]

         val
id time     
1  2       3
2  2       4