Pandas中怎么分组再取N项

本篇内容介绍了“Pandas中怎么分组再取N项”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

成都创新互联专注于无极网站建设服务及定制,我们拥有丰富的企业做网站经验。 热诚为您提供无极营销型网站建设,无极网站制作、无极网页设计、无极网站官网定制、重庆小程序开发公司服务,打造无极网络公司原创品牌,更为您提供无极网站排名全网营销落地服务。

在 pandas 中,DataFrame 是我们经常用到的工具。有时候,我们可能会需要对数据按某个字段进行分组,然后每个组取N项。例如:

Pandas中怎么分组再取N项

现在,我想每个职位任取三个用户。

相信有同学会使用 for 循环,依次循环每一行,每个职位选3个,存入一个临时的列表里面。循环完成以后再转成一个新的  DataFrame。但这个方式显然不够智能。

那么,我们有没有什么办法能够不使用循环就做到这一步呢?也许有同学想到了使用 groupby。我们来看看效果。

Pandas中怎么分组再取N项

看起来仅仅是统计了每个职位的数量。那么,如何才能保留所有字段呢?

实际上我们可以把.size()改成.head(3):

Pandas中怎么分组再取N项

看起来这里的.head(3)似乎没有什么作用。这个时候,我们思考一下 Python  里面,如果要使用itertools.groupby,官方文档里面有这样一段话:

Generally, the iterable needs to already be sorted on the same key  function.

如下图所示:

Pandas中怎么分组再取N项

这段话告诉我们,要使用itertools.groupby,我们需要提前对被分组的字段进行排序。

那么,我们试一试在如果提前对 DataFrame 进行排序,然后再 groupby 会怎么样:

Pandas中怎么分组再取N项

成功了。每个职位都取了3个。

可能大家发现最左边的索引是乱序,看起来不好看。那么我们还可以重设一下索引:

Pandas中怎么分组再取N项

“Pandas中怎么分组再取N项”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注创新互联网站,小编将为大家输出更多高质量的实用文章!


标题名称:Pandas中怎么分组再取N项
分享网址:http://azwzsj.com/article/gdecis.html