python - 如何使用 Python Pandas 根据组中其他列的聚合来选择值？

Question

我的数据如下所示:

system    question    answer    grade    rank
sys1      q1          a1        A        5
sys1      q1          a1        B        10
sys2      q1          a1        C        1
sys2      q1          a1        D        11

我的目标是按问题和答案进行分组，取最好的成绩和排名，以及与最好成绩相关的系统。下面是我想要执行的示例:

>>df.groupby(['question', 'answer']).agg({'grade': np.min, 'rank': np.min, 'system': ???).reset_index()
question    answer    grade    rank    system
q1          a1        A        1       sys1

我得到了我想要的成绩和排名，因为我只需要最低限度。获得所需系统的最佳方式是什么？

Answer 1

我认为你可以使用sort_values先聚合last :

print (df)
  system question answer grade  rank
0   sys1       q1     a1     A     5
1   sys2       q1     a1     B    10
2   sys3       q1     a1     C     1
3   sys4       q1     a1     D    11

df = df.sort_values('grade')
       .groupby(['question', 'answer'])
       .agg({'grade': np.min, 'rank': np.min, 'system':'last'})
       .reset_index()
print (df)
  question answer system  rank grade
0       q1     a1   sys4     1     A

<小时/>

我不确定，但也许需要首先:

df = df.sort_values('grade')
       .groupby(['question', 'answer'])
       .agg({'grade': np.min, 'rank': np.min, 'system':'first'})
       .reset_index()
print (df)
  question answer system  rank grade
0       q1     a1   sys1     1     A