[pandas] 데이터프레임 열 또는 행 기준으로 요소 총합 구하기, sum() 메서드

판다스 데이터프레임에서 열 또는 행 기준으로 요소의 총합을 구할 때는 sum() 메서드를 사용할 수 있습니다. 우선 예시를 위해 저는 영수, 철수, 광수의 수학, 영어, 과학 점수를 담은 데이터프레임 df1을 하나 생성했습니다. 

 

import pandas as pd


df1 = pd.DataFrame(
    [[90, 85, 80], [77, 84, 79], [86, 89, 91]],
    columns=["수학", "영어", "과학"],
    index=["영수", "철수", "광수"])

print(df1)
#     수학  영어  과학
# 영수  90  85  80
# 철수  77  84  79
# 광수  86  89  91

print(df1.sum(axis=0))  # 열 요소들의 총합
# 수학    253
# 영어    258
# 과학    250
# dtype: int64

print(df1.sum(axis=1))  # 행 요소들의 총합
# 영수    255
# 철수    240
# 광수    266
# dtype: int64

 

보시다시피 axis=0으로 설정하면 열 요소들의 총합을 구하고, axis=1로 설정하면 행 요소들의 총합을 구합니다.