[파이썬 numpy] 배열의 누적합 (axis 방향별)

[파이썬 numpy] 

배열의 누적합 (axis 방향별)

배열의 누적합을 구할때는 cumsum 메소드를 사용합니다.  

먼저 배열을 하나 정의합시다.

>>> import numpy as np

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

배열의 누적합

배열의 누적합은 아래와 같이 두 가지 방법으로 구할 수 있습니다. 넘파이 함수를 사용해도 되고, 정의한 배열에 메소드를 적용해도 됩니다. 

>>> np.cumsum(A)

array([ 1,  3,  6, 10, 15], dtype=int32)

>>> A.cumsum()

array([ 1,  3,  6, 10, 15], dtype=int32)

배열의 axis 에 따른 누적합

2차원 배열을 하나 정의합시다.

>>> A=np.arange(6).reshape(2,3)

>>> A

array([[0, 1, 2],

       [3, 4, 5]])

axis=0 방향의 누적합을 구해봅시다. 행방향 누적합입니다. 

>>> A.cumsum(axis=0)

array([[0, 1, 2],

       [3, 5, 7]], dtype=int32)

axis=1 방향의 누적합을 구해봅시다. 열방향 누적합입니다. 

>> A.cumsum(axis=1)

array([[ 0,  1,  3],

       [ 3,  7, 12]], dtype=int32)