python - 在python中记录每个id的最大系列

Question

我想为每个 id 保留一个具有最大系列的记录。所以对于每个 id 我需要一行。我想我需要类似的东西

df_new = df.groupby('id')['series'].nlargest(1)

，但这绝对是错误的。

这就是我的数据集的样子:

id  series s1 s2 s3
1   2      4  9  1
1   8      6  2  2
1   3      9  1  3
2   9      4  1  5
2   2      2  5  5
2   5      1  7  8
3   6      7  2  3
3   2      4  4  1
3   1      3  9  9

应该是这样的结果:

id  series s1 s2 s3
1   8      6  2  2
2   9      4  1  5
3   6      7  2  3

Answer 1

IIUC你要groupby在“id”列上，使用 idxmax() 获取“Series”值最大的索引标签并使用它在 orig df 中返回索引:

In [91]:
df.loc[df.groupby('id')['series'].idxmax()]

Out[91]:
   id  series  s1  s2  s3
1   1       8   6   2   2
3   2       9   4   1   5
6   3       6   7   2   3