python - Pandas 分组并根据条件添加列数据

Question

我合并了如下测试数据:

Device       time    Key score
Computers 2018-01-01 14.0 4.0
Computers 2018-01-01 11.0 4.0
Computers 2018-01-01 16.0 0.0

我需要按列 [Device,time] 和列分数的最大值对数据进行分组，并获得分配给该分数的最小键值。

我的第 1 次尝试:

df_out = df_out.groupby(['Device', 'time'])['score'].max().reset_index()

输出 1:

Device       time    score
Computers 2018-01-01 4.0

我的第二次尝试:

df_out = df_out.groupby(['Device', 'time'])['score', 'Key'].max().reset_index()

输出 2:

Device       time    score Key
Computers 2018-01-01 4.0  14.0

如何分配合适的最小Key？

期望的输出:

Device       time    score Key
Computers 2018-01-01 4.0  11.0

谢谢你的辛勤工作。

Answer 1

您可以使用转换:

df[df.score.eq(df.groupby(['Device', 'time'])['score'].transform('max'))]

---

      Device        time   Key  score
0  Computers  2018-01-01  14.0    4.0

根据编辑:

df.groupby(['Device', 'time'],as_index=False).agg({'score':'max','Key':'min'})

---

      Device        time  score   Key
0  Computers  2018-01-01    4.0  11.0