python - 对列进行排序并过滤最佳结果，按多索引级别 0(系列/数据帧)分组

Question

我想将索引中每个元素的最高结果分组为 0 级。例如使用此数据框/系列:

import pandas as pd
import numpy as np
np.random.seed(1)

index = list(zip(['A']*5 + ['B']*5, list(range(10))))
df = pd.Series(np.random.random((10)),
               index=pd.MultiIndex.from_tuples(index, names=['i0', 'i1']),
               name='val')

pd.DataFrame(df)

我想将 A 和 B 分组并返回前 3 个 val(降序)。

Answer 1

这里还有一个类似的选项:

(df.sort_values('val', ascending=False).groupby(level=0).head(3)
   .sort_index(level = 0, sort_remaining=False, kind="mergesort"))