pandas groupby 分组取每组的前几行记录

直接上例子。

import pandas as pd
df = pd.DataFrame({'class':['a','a','b','b','a','a','b','c','c'],'score':[3,5,6,7,8,9,10,11,14]})

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
pandasgroupby函数是对数据进行分组操作的一种常用方法。它可以将数据按照一列或多列进行分组,并对每个组进行聚合、转换或其他操作。当需要对分组后的数据进行筛选,只取前几行时,可以使用head()函数。 在groupby函数中,通过传入多个列名可以按照多列进行分组。例如,假设我们有一个包含学生姓名、学科和成绩的数据集,我们想按照学科和姓名这两列进行分组,并每个组的几行数据。 假设我们已经将数据导入pandas的DataFrame中,可以这样操作: ``` python import pandas as pd # 创建DataFrame data = {'姓名': ['张三', '李四', '王', '赵六', '刘七', '张三', '李四', '王', '赵六', '刘七'], '学科': ['语文', '数学', '语文', '数学', '英语', '语文', '数学', '语文', '数学', '英语'], '成绩': [90, 80, 70, 60, 50, 85, 75, 65, 55, 45]} df = pd.DataFrame(data) # 按照学科和姓名进行分组,并取前两行数据 top2 = df.groupby(['学科', '姓名']).head(2) print(top2) ``` 输出结果为: ``` 姓名 学科 成绩 0 张三 语文 90 1 李四 数学 80 2 王 语文 70 3 赵六 数学 60 4 刘七 英语 50 5 张三 语文 85 6 李四 数学 75 7 王 语文 65 8 赵六 数学 55 ``` 通过groupby函数按照学科和姓名进行分组后,再使用head(2)函数每个组的两行数据。结果中会包含每个学科和姓名组合的两行数据。 总结起来,使用pandasgroupby函数按照多列进行分组,再用head()可方便地取前几行数据。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值