python - 对于a列中的相同值，对b列中的值进行排序并选择前5个

Question

我有一个包含两列 col_1 和 col_2 的数据框。 col_2 列中的值对应于 col_1 列中的值。

print (df)
   col_1  col_2
1      a     12
2      a     33
3      a     11
4      a      4
5      a     42
6      a     66
7      a      9
8      b     12
9      b     34
10     b     42
11     b     64
12     b     86
13     b      2

我想做的是，对于 col_1 (a, b, c,..) 中的每个值，我想对 中的相应值进行排序code>col_2 并仅选择前 5 个值。新的数据框应该是这样的:

我尝试了删除重复应用 here因为 col_2 有时会重复。但是，它没有用。

df.sort_values('col_2', ascending=False).drop_duplicates('col_a').sort_index()

任何建议将不胜感激

Answer 1

改用 drop_duplicates 函数 GroupBy.head并将第一列添加到 DataFrame.sort_values升序排序:

n = 5
df1 = df.sort_values(['col_1','col_2'], ascending=[True, False]).groupby('col_1').head(n)
print (df1)
   col_1  col_2
6      a     66
5      a     42
2      a     33
1      a     12
3      a     11
12     b     86
11     b     64
10     b     42
9      b     34
8      b     12