Python Pandas – 데이터프레임의 모든 행 합계 구하는 방법

Pandas 데이터프레임(DataFrame)에서 모든 행의 합계를 구하려면 sum() 함수를 사용하고 axis=1로 설정하면 됩니다. axis를 1로 지정하면 각 행의 값들이 가로 방향으로 더해져 행별 합계가 계산됩니다.

데이터프레임 생성하기

먼저 예제에 사용할 데이터프레임을 만들어 보겠습니다. 이 데이터프레임에는 'Opening_Stock'(기초 재고)과 'Closing_Stock'(기말 재고) 두 개의 열이 포함되어 있습니다.

dataFrame = pd.DataFrame({
    "Opening_Stock": [300, 700, 1200, 1500],
    "Closing_Stock": [200, 500, 1000, 900]
})

행 합계 구하기

각 행의 값들을 더하기 위해 sum() 함수를 호출하면서 axis를 1로 설정합니다.

dataFrame = dataFrame.sum(axis = 1)

전체 예제 코드

다음은 위 과정을 정리한 완전한 코드입니다.

import pandas as pd

dataFrame = pd.DataFrame({
    "Opening_Stock": [300, 700, 1200, 1500],
    "Closing_Stock": [200, 500, 1000, 900]
})

print("DataFrame...\n", dataFrame)

# 행 값들의 합계 구하기
# axis를 1로 설정하여 행 값을 더함
dataFrame = dataFrame.sum(axis = 1)
print("\nSumming rows...\n", dataFrame)

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다.

DataFrame...
   Closing_Stock  Opening_Stock
0            200            300
1            500            700
2           1000           1200
3            900           1500

Summing rows...
0     500
1    1200
2    2200
3    2400
dtype: int64

결과 해석 및 참고 사항

실행 결과를 보면 첫 번째 행은 300 + 200 = 500, 두 번째 행은 700 + 500 = 1200처럼 각 행의 Opening_Stock과 Closing_Stock 값이 더해진 것을 확인할 수 있습니다.

참고로 axis=0(기본값)으로 설정하면 열 단위 합계가 계산됩니다. 따라서 행 단위 합계를 구하고 싶다면 반드시 axis=1을 지정해야 한다는 점을 기억하세요.