numpy.cumsum¶

numpy.cumsum 함수는 지정된 축을 따라 어레이 요소의 누적 합을 계산합니다.

예제1¶

import numpy as np

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

print(a)
print(np.cumsum(a))

[[1 2 3]
[4 5 6]]
[ 1  3  6 10 15 21]

np.cumsum() 함수는 어레이 요소를 따라 누적 합을 계산합니다.

axis를 지정하지 않았기 때문에 flattened 어레이에 대해 계산합니다.

import numpy as np

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

print(np.cumsum(a, axis=0))
print(np.cumsum(a, axis=1))

[[1 2 3]
[5 7 9]]
[[ 1  3  6]
[ 4  9 15]]

np.cumsum() 함수에 axis를 지정하면 지정한 축을 따라 계산합니다.

axis=0으로 지정하면 첫번째 축을 따라,

axis=1으로 지정하면 두번째 축을 따라 누적 합을 반환합니다.

이전글 : numpy.cosh