我得到像这样的熊猫数据帧：
id prob
0 1 0.5
1 1 0.6
2 1 0.4
3 1 0.2
4 2 0.3
6 2 0.5
...
我想按’id’对它进行分组,按降序排序并获得每组的前3个概率.请注意,某些组包含小于3的行.
最后我想得到一个2D数组：
[[1, 0.6, 0.5, 0.4], [2, [0.5, 0.3]]...]
我怎么能用熊猫做到这一点？
谢谢！
解决方法:
使用sort_values,groupby和head：
df.sort_values(by=['id','prob'], ascending=[True,False]).groupby('id').head(3).values
输出：
array([[ 1. , 0.6],
[ 1. , 0.5],
[ 1. , 0.4],
[ 2. , 0.5],
[ 2. , 0.3]])
关注@COLDSPEED领导：
df.sort_values(by=['id','prob'], ascending=[True,False])\
.groupby('id').agg(lambda x: x.head(3).tolist())\
.reset_index().values.tolist()
输出：
[[1, [0.6, 0.5, 0.4]], [2, [0.5, 0.3]]]
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 [email protected] 举报，一经查实，本站将立刻删除。
