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

[파이썬 numpy] 

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

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

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

>>> import numpy as np

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

배열의 누적곱

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

>>> np.cumprod(A)

array([  1,   2,   6,  24, 120], dtype=int32)

>>> A.cumprod()

array([  1,   2,   6,  24, 120], dtype=int32)

배열의 axis 에 따른 누적곱

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

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

>>> A

array([[1, 2, 3],

       [4, 5, 6]])

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

>>> A.cumprod(axis=0)

array([[ 1,  2,  3],

       [ 4, 10, 18]], dtype=int32)

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

>>> A.cumprod(axis=1)

array([[  1,   2,   6],

       [  4,  20, 120]], dtype=int32)