python - 在 Pandas 数据框中选择得分最高的行

Question

我有一个 pandas 数据框，它看起来像:

Name Candidates Qualifier Score
AAA   AAA_1       Yes      0
AAA   AAA_2       Yes      10
AAA   AAA_3        No      20
BBB   BBB_1        No      1
BBB   BBB-2       Yes      10
BBB   BBB_3       Yes     50

我想在每个“名称”中选择得分最高的前两名候选人。我该怎么做？

Answer 1

您可以对 'Name' 和 'Score' 的 df 进行排序，然后对 'Name' 进行 groupby 并调用 head(2) 以获得前 2 行每组:

In [228]:
df.sort(['Name','Score'], ascending=False).groupby('Name').head(2)

Out[228]:
  Name Candidates Qualifier  Score
5  BBB      BBB_3       Yes     50
4  BBB      BBB-2       Yes     10
2  AAA      AAA_3        No     20
1  AAA      AAA_2       Yes     10