python - 如何在通用级别 l 切片 pandas.MultiIndex？

Question

df.xs(i, level=l) 获取索引 i 在级别 l 的值，但不允许设置它们。

以下设置索引 i 在级别 2 的值 (cf. documentation) :

df.loc[pd.IndexSlice[:,:,i],:] = 0

有没有办法在级别 l(对于任何级别 l)设置索引 i 的值？我需要这个，因为我事先不知道我需要在什么水平上操作。

Answer 1

我想你需要 pd.index.get_level_values

import pandas as pd
import numpy as np

np.random.seed(123)
df = pd.DataFrame(np.random.randint(1,50,(100,8))).set_index(list(range(5)))

i = 7

l = 2
df.loc[df.index.get_level_values(l) == i]
#               5   6   7
#0  1  2 3  4            
#15 12 7 36 24  6  48  41

l = 4
df.loc[df.index.get_level_values(l) == i]
#                5  6   7
#0  1  2  3  4           
#22 31 21 32 7  33  8  14