python - pandas [3] groupby

0. groupby 메서드

- 데이터를 그룹 별로 분류하는 역할

 

import pandas as pd
import numpy as np
np.random.seed(0)
df = pd.DataFrame({
    'key1': ['A', 'A', 'B', 'B', 'A'],
    'key2': ['one', 'two', 'one', 'two', 'one'],
    'data1': [1, 2, 3, 4, 5],
    'data2': [10, 20, 30, 40, 50]
})
 
print(df)
"""
  key1 key2  data1  data2
0    A  one      1     10
1    A  two      2     20
2    B  one      3     30
3    B  two      4     40
4    A  one      5     50
"""
 
# key1을 기준
groups = df.groupby(df.key1)
 
#속성
print(groups.groups)
"""
{'A': Int64Index([0, 1, 4], dtype='int64'), 'B': Int64Index([2, 3], dtype='int64')}
"""
 
#data1에서만 그룹연산 하는 코드
print(df.data1.groupby(df.key1).sum())
"""
key1
A    8
B    7
Name: data1, dtype: int64
"""
 
print(df.groupby(df.key1)["data1"].sum())
"""
key1
A    8
B    7
Name: data1, dtype: int64
"""
 
print(df.groupby(df.key2).sum()["data2"] )
"""
key2
one    90
two    60
Name: data2, dtype: int64
"""
 
 
 
 

 

 

<출처> 

1. 데이터 사이언스 스쿨