[Python/파이썬] numpy.sum(), 배열의 합구하기

예제 1: 1차원 배열의 합

import numpy as np

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

1차원 배열의 모든 요소를 합하는 예제입니다. numpy의 sum() 함수를 사용하여 배열의 합을 계산합니다.

예제 2: 다차원 배열의 합

import numpy as np

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

다차원 배열의 모든 요소를 합하는 예제입니다. sum() 함수를 사용하여 다차원 배열의 합을 계산합니다.

예제 3: 축을 지정하여 합 계산

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
sum_result_axis0 = np.sum(arr, axis=0)
print(sum_result_axis0)

다차원 배열에서 특정 축을 기준으로 합을 계산하는 예제입니다. axis 매개변수를 사용하여 합을 계산할 축을 지정합니다.

예제 4: keepdims 매개변수 사용

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
sum_result_keepdims = np.sum(arr, axis=0, keepdims=True)
print(sum_result_keepdims)

keepdims 매개변수를 사용하여 출력 결과의 차원을 유지하는 예제입니다. keepdims를 True로 설정하면 출력 결과의 차원이 입력과 동일하게 유지됩니다.

예제 5: 배열의 특정 요소 합

import numpy as np

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

다차원 배열에서 특정 축을 기준으로 합을 계산하는 예제입니다. axis 매개변수를 사용하여 합을 계산할 축을 지정합니다.

예제 6: dtype 매개변수 사용

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
sum_result_dtype = np.sum(arr, dtype=np.float64)
print(sum_result_dtype)

dtype 매개변수를 사용하여 합을 계산할 때 결과의 데이터 타입을 지정하는 예제입니다. 결과의 데이터 타입을 정확하게 제어할 수 있습니다.