Python/Pandas

Pandas – groupby를 사용하여 특정열 기준으로 DataFrame의 데이터 그룹 나누기

webnautes 2023. 10. 9. 10:50
반응형

groupby를 사용하여 특정열 기준으로 DataFrame의 데이터 그룹 나누는  Pandas 예제 코드입니다.



2022. 10. 21  최초작성



import pandas as pd


# DataFrame은 2개의 컬럼으로 구성되며 alphabet 컬럼의 값을 기준으로  그룹을 나누려고 합니다. 
data = pd.DataFrame([[1, 'a'],[2, 'a'],[3,'b'],[4,'b'],[5,'b'],[4,'c']], columns=['number', 'alphabet'])
print(data)
  number alphabet
0       1        a
1       2        a
2       3        b
3       4        b
4       5        b
5       4        c


# alphabet 컬럼의 값을 기준으로 그룹으로 묶을 수 있도록 합니다.
groups = data.groupby('alphabet')


# alphabet 컬럼의 값이 a인 행만 가져옵니다. 
group_a = groups.get_group('a')

# alphabet 컬럼의 값이 b인 행만 가져옵니다. 
group_b = groups.get_group('b')

# alphabet 컬럼의 값이 a인 행만 출력됩니다.
print(group_a)
  number alphabet
0       1        a
1       2        a

# alphabet 컬럼의 값이 b인 행만 출력됩니다.
print(group_b)
  number alphabet
2       3        b
3       4        b
4       5        b



반응형