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

Numpy Cumulative Sum 함수 소개:

Numpy의 `cumsum` 함수는 배열의 누적 합을 계산하는 데 사용됩니다. 이 함수는 주어진 축을 따라 각 요소의 누적 합을 계산하여 반환합니다.

기본 사용법:

`cumsum` 함수를 사용하여 배열의 누적 합을 계산하는 방법은 다음과 같습니다.

import numpy as np

# 누적 합을 계산할 배열 생성
arr = np.array([1, 2, 3, 4, 5])

# 누적 합 계산
cumulative_sum = np.cumsum(arr)

print("누적 합 결과:", cumulative_sum)

위의 예시 코드에서는 주어진 배열에 대해 `cumsum` 함수를 사용하여 누적 합을 계산합니다.

추가 옵션:

`cumsum` 함수는 추가적인 옵션을 지원합니다. 예를 들어, 축을 지정하여 해당 축을 따라 누적 합을 계산할 수 있습니다.

# 다차원 배열에 대한 누적 합 계산 (축 지정)
arr = np.array([[1, 2, 3], [4, 5, 6]])
cumulative_sum_axis0 = np.cumsum(arr, axis=0)
cumulative_sum_axis1 = np.cumsum(arr, axis=1)

print("축 0을 따른 누적 합 결과:", cumulative_sum_axis0)
print("축 1을 따른 누적 합 결과:", cumulative_sum_axis1)

위의 코드에서는 다차원 배열에 대해 축을 지정하여 누적 합을 계산하는 방법을 보여줍니다.

결과:

`cumsum` 함수는 배열의 누적 합을 계산하는 데 사용되며, 다양한 옵션을 통해 유연하게 적용할 수 있습니다.