Pandas GroupBy grouping takes the first few rows of each group record method

The following for everyone to share a pandas GroupBy group to take the first few lines of the record method, with a good reference value, I hope to be helpful to everyone. Come and see it together.

Directly on the example.

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