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

下面为大家分享一篇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]})

df:

	class	score
0	a	3
1	a	5
2	b	6
3	b	7
4	a	8
5	a	9
6	b	10
7	c	11
8	c	14

df.sort_values(['class','score'],ascending=[1,0],inplace=True) 
grouped = df.groupby(['class']).head(2)

grouped:

	class	score
5	a	9
4	a	8
6	b	10
3	b	7
8	c	14
7	c	11

相关推荐：

pandas多级分组实现排序的方法

pandas获取groupby分组里最大值所在的行方法

Python 数据处理库 pandas 入门

以上是pandas groupby 分组取每组的前几行记录方法的详细内容。更多信息请关注PHP中文网其他相关文章！