一日一技:Pandas 中,如何分组再取 N项?
2023-03-14 09:38:48 时间
在 pandas 中,DataFrame 是我们经常用到的工具。有时候,我们可能会需要对数据按某个字段进行分组,然后每个组取N项。例如:
现在,我想每个职位任取三个用户。
相信有同学会使用 for 循环,依次循环每一行,每个职位选3个,存入一个临时的列表里面。循环完成以后再转成一个新的 DataFrame。但这个方式显然不够智能。
那么,我们有没有什么办法能够不使用循环就做到这一步呢?也许有同学想到了使用 groupby。我们来看看效果。
看起来仅仅是统计了每个职位的数量。那么,如何才能保留所有字段呢?
实际上我们可以把.size()改成.head(3):
看起来这里的.head(3)似乎没有什么作用。这个时候,我们思考一下 Python 里面,如果要使用itertools.groupby,官方文档里面有这样一段话:
Generally, the iterable needs to already be sorted on the same key function.
如下图所示:
这段话告诉我们,要使用itertools.groupby,我们需要提前对被分组的字段进行排序。
那么,我们试一试在如果提前对 DataFrame 进行排序,然后再 groupby 会怎么样:
成功了。每个职位都取了3个。
可能大家发现最左边的索引是乱序,看起来不好看。那么我们还可以重设一下索引:
至此,问题完美解决。
本文转载自微信公众号「未闻Code」,可以通过以下二维码关注。转载本文请联系未闻Code公众号。
相关文章
- 如何在 Kubuntu 21.10 中升级 KDE Plasma 5.24
- Google 新操作系统 Chrome OS Flex 诞生
- Linux中排除网络故障时,这些Linux命令必须要回!
- VSCode 阅读 Linux 代码怎么才不卡顿?这样做才能快的飞起!
- Safari 浏览器新变,苹果 iOS 15.4 Beta iCloud 钥匙串不再保存无用户名的密码
- 这年头 Linux 竟然成了程序员标配
- 苹果 macOS 12.3 开发者预览版 Beta 3 优化通用控制功能,设置更方便
- 一日一技:如何在浏览器中使用 Npm包?
- HarmonyOS-ETS之紧急拨号
- HarmonyOS-Java之手把手教你绘制冰墩墩
- 润和Hi3516开发板OpenHarmony小型系统、标准系统烧录
- 【FFH】小熊派驱动调用流程(以调用LED灯驱动为例)
- 服务器性能优化之网络性能优化
- Preset-Env 按需 Polyfill 是怎么实现的?
- Windows不安装虚拟机如何使用Linux系统作为开发工具?
- 一款运行于Windows上的Linux命令神器-Cmder(已经爱不释手)
- 谷歌推出 Chrome OS Flex 系统,可将旧电脑变成 Chromebook
- Wine 7.2 发布,Linux下安装使用最新版QQ与微信
- 直观对比了解Windows 10和Windows 11,才知道最好用的系统是这个
- Windows 11任务栏有硬伤:微软终于要改了 功能回归正常