Python numpy.sum(), 배열의 합

pythonnumpy

numpy.sum()을 사용하여 numpy 배열의 합을 계산하는 방법을 소개합니다.

1. 1차 배열의 합 계산

아래와 같이 1차 배열의 합을 계산할 수 있습니다.

  • numpy.sum(arr)은 arr의 모든 요소의 합을 리턴
  • initial 인자는 초기 값으로, 합을 계산할 때 배열의 모든 요소와 initial 값을 더함
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
total = np.sum(arr)
print(total)

arr = np.array([1, 2, 3, 4, 5])
total = np.sum(arr, initial=10)
print(total)

Output:

15
25

2. 2차 배열의 합 계산

아래와 같이 2차 배열의 합을 계산할 수 있습니다.

  • numpy.sum(arr)은 arr의 모든 요소의 합을 리턴
  • initial 인자를 사용하여 초기 값을 설정할 수 있음
import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]])
total = np.sum(arr)
print(total)

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
total = np.sum(arr)
print(total)

Output:

55
45

3. 2차 배열에서 행 또는 열의 합 계산

sum()에 axis 인자를 전달하여, 2차 배열에서 같은 행 또는 열에 속하는 요소들의 합만 계산할 수 있습니다.

  • axis=0은 같은 열의 요소들 끼리 합을 계산
  • axis=1은 같은 행의 요소들 끼리 합을 계산
import numpy as np

# 모든 배열 요소들의 합 계산
arr = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]])
print(arr)

# 같은 열의 요소들 끼리 합 계산
total = np.sum(arr, axis=0)
print(f"sum of column: {total}")

# 같은 행의 요소들끼리 합 계산
total = np.sum(arr, axis=1)
print(f"sum of row: {total}")

결과를 보면, 첫번째는 1열과 2열의 요소들끼리 합을 계산하여 배열로 리턴하였습니다. 두번째는 1~5행의 각 요소들끼리 합을 계산하여 배열로 리턴하였습니다.

[[ 1  2]
 [ 3  4]
 [ 5  6]
 [ 7  8]
 [ 9 10]]
sum of column: [25 30]
sum of row: [ 3  7 11 15 19]

4. 특정 조건을 충족하는 요소들의 합 계산

모든 요소들의 합을 계산하지 않고, 요소들 중에 특정 조건을 만족하는 경우만 찾아서 합을 계산할 수 있습니다.

  • numpy.sum(arr)은 모든 요소의 합 계산
  • numpy.sum(arr[condition])은 condition이 true를 리턴하는 요소들의 합 계산
  • numpy.sum(arr[arr > 3])은 요소 값이 3을 초과하는 요소들의 합 계산
import numpy as np

arr = np.array([1, 2, 3, 4, 5])

total = np.sum(arr[arr > 3])
print(total)

total = np.sum(arr[arr <= 3])
print(total)

Output:

9
6
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha