如何根据 MultiIndex 的一个级别上的条件选择 DataFrame 的子集

Question

类似于如何根据MultiIndex的一个级别选择DataFrame的子集，让

df = pd.DataFrame({"v":[x*x for x in range(12)]}, 
                  index=pd.MultiIndex.from_product([["a","b","c"],[1,2,3,4]]))

假设我只想选择 second 级别与给定 first 级别的最小值相差 25 以内的行：

       v
a 1    0
  2    1
  3    4
  4    9
b 1   16
  2   25
  3   36
c 1   64
  2   81

这一次我不知道如何轻松地做到这一点（除了

to_frame

和

groupby

的复杂组合）......

Answer 1

您可以按数据框的级别 0 进行分组，并获得每个组中

v

列的最小值。比较每组中的

v

和最小的

v

。

out = df[df['v'].sub(df.groupby(level=0)['v'].transform('min')) < 25]

print(out)

      v
a 1   0
  2   1
  3   4
  4   9
b 1  16
  2  25
  3  36
c 1  64
  2  81