Pandas Multiindex數據幀刪除行

Question

我有Multiiindex DF如下：

tuples = list(zip(*[['a', 'a', 'b', 'b'], ['c', 'd', 'c', 'd']]))
index = pd.MultiIndex.from_tuples(tuples, names=['i1', 'i2'])
df = pd.DataFrame([5, 6, 7, 8], index=index[:4], columns=['col'])

       col
i1 i2     
a  c     5
   d     6
b  c     7
   d     8

想保留索引（級別0）所在的行

idx_to_keep = ['a']

應該是一個簡單的任務，但我想不出任何其他方式

idx_to_drop = np.setdiff1d(pd.unique(df.index.levels[0]), idx_to_keep)
df.drop(idx_to_drop, inplace = True)

       col
i1 i2     
a  c     5
   d     6

我可以做得更好嗎？

Answer 1

一種方法是使用index方法get_level_values() ：

df
       col
i1 i2     
a  c     5
   d     6
b  c     7
   d     8

df[df.index.get_level_values(0).isin(idx_to_keep)]
       col
i1 i2     
a  c     5
   d     6

Answer 2

您正在尋找.xs ：

df.xs('a', axis=0, level=0, drop_level=False)

這使：

       col
i1 i2     
a  c     5
   d     6

Answer 3

你可以使用loc ：

df.loc[['a']]

結果輸出：

       col
i1 i2     
a  c     5
   d     6

Answer 4

我們來使用slice

idx_to_keep = ['a']
df.loc[slice(*idx_to_keep,)]

輸出：

       col
i1 i2     
a  c     5
   d     6