본문 바로가기
Python/numpy

파이썬 numpy.sum 함수 활용하기

by PySun 2024. 6. 15.
반응형

Numpy Sum 함수 소개:

Numpy의 `numpy.sum` 함수는 배열의 원소들을 합하는 함수로, 주어진 배열의 모든 원소를 더하여 총합을 계산합니다. 이 함수는 다차원 배열에서도 축(axis)을 지정하여 합을 계산할 수 있어서 데이터 처리 및 분석에 유용하게 활용됩니다.

기본 사용법:

`numpy.sum` 함수의 기본 사용법은 다음과 같습니다.

import numpy as np

# 1차원 배열의 모든 원소의 합 계산
array = np.array([1, 2, 3, 4, 5])
total_sum = np.sum(array)
print("배열의 총합:", total_sum)

위 예시 코드에서 `np.sum(array)`는 주어진 1차원 배열의 모든 원소의 합을 계산합니다.

예시 코드:

  • 다차원 배열의 합 계산:
# 다차원 배열의 합 계산
array_2d = np.array([[1, 2], [3, 4]])
array_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

sum_2d = np.sum(array_2d)  # 모든 원소의 합
sum_2d_axis_0 = np.sum(array_2d, axis=0)  # 각 열의 합
sum_2d_axis_1 = np.sum(array_2d, axis=1)  # 각 행의 합

sum_3d = np.sum(array_3d)  # 모든 원소의 합
sum_3d_axis_0 = np.sum(array_3d, axis=0)  # 각 깊이 차원(axis=0)의 합
sum_3d_axis_1 = np.sum(array_3d, axis=1)  # 각 행의 합
sum_3d_axis_2 = np.sum(array_3d, axis=2)  # 각 열의 합

print("2차원 배열 총합:", sum_2d)
print("2차원 배열 열별 합:", sum_2d_axis_0)
print("2차원 배열 행별 합:", sum_2d_axis_1)

print("3차원 배열 총합:", sum_3d)
print("3차원 배열 깊이별 합:", sum_3d_axis_0)
print("3차원 배열 행별 합:", sum_3d_axis_1)
print("3차원 배열 열별 합:", sum_3d_axis_2)

결론:

`numpy.sum` 함수는 다차원 배열의 합을 계산하는 유용한 함수로, 데이터 처리 및 분석 작업에서 자주 활용됩니다. 다양한 축(axis)을 지정하여 배열의 합을 계산할 수 있어서 데이터의 특성을 다양하게 분석할 수 있습니다. 예시 코드를 통해 `numpy.sum` 함수의 활용법을 익혀보세요.

반응형