Pandas:각 그룹의 첫 번째 행을 얻는 방법

다음 기본 구문을 사용하여 Pandas DataFrame에서 각 그룹의 첫 번째 행을 가져올 수 있습니다.

 df. groupby (' column_name '). nth ( 0 )

다음 예에서는 실제로 이 구문을 사용하는 방법을 보여줍니다.

예: Pandas에서 각 그룹의 첫 번째 행 가져오기

다음과 같은 팬더 DataFrame이 있다고 가정합니다.

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 29],
                   ' assists ': [5, 19, 14, 8, 9, 12, 13, 8]})

#view DataFrame
df

	team points assists
0 to 18 5
1 To 22 19
2 B 19 14
3 B 14 8
4 B 14 9
5 C 11 12
6 C 20 13
7 C 29 8

다음 코드를 사용하여 각 팀의 첫 번째 행을 가져올 수 있습니다.

 #get first row for each team
df. groupby (' team '). nth ( 0 )

	assist points
team		
At 18 5
B 19 14
C 11 12

as_index=False를 지정하여 원래 인덱스 값을 유지할 수도 있습니다.

 #get first row for each team, keep original index values
df. groupby (' team ', as_index= False ). nth ( 0 )

        team points assists
0 to 18 5
2 B 19 14
5 C 11 12

또한 각 그룹의 처음 n 행을 얻으려면 nth() 함수에 값 목록을 전달할 수 있습니다.

예를 들어 다음 코드는 각 그룹의 처음 두 행을 가져오는 방법을 보여줍니다.

 #get first two rows for each team, keep original index values
df. groupby (' team ', as_index= False ). nth (( 0,1 ) )

team points assists
0 to 18 5
1 To 22 19
2 B 19 14
3 B 14 8
5 C 11 12
6 C 20 13

참고 : nth() 함수에 대한 전체 문서는 여기에서 찾을 수 있습니다.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

Pandas DataFrame의 첫 번째 행을 얻는 방법
Pandas DataFrame에서 첫 번째 행을 제거하는 방법
Pandas DataFrame에 행을 삽입하는 방법