pandas - 按索引号排序的 MultiIndex DataFrame 索引

Question

我有一个 MultiIndex DataFrame 如下:

header = pd.MultiIndex.from_product([['#'],
                                     ['TE', 'SS', 'M', 'MR']])
dat = ([[100, 20, 21, 35], [100, 12, 5, 15]])
df = pd.DataFrame(dat, index=['JC', 'TTo'], columns=header)
df = df.stack()
df = df.sort_values('#', ascending=False).sort_index(level=0, sort_remaining=False)

我想通过索引号而不是名称获取下一行索引，即每个 0 级索引的第三行:

JC   M   21
TTo  SS  12

在我尝试过的所有方法中，最接近我正在寻找的是:

df.loc[pd.IndexSlice[:, df.index[2]], '#']

但这也不能按预期工作。

Answer 1

您可以执行以下操作:

df["idx"] = df[df.groupby(level=0).cumcount() == 2]
df.loc[df.idx == 2]

Quang Hoang 的一线解决方案:

df[df.groupby(level=0).cumcount() == 2]