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

[파이썬 numpy] 

배열의 합 (axis 방향별)

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

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

>>> import numpy as np

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

배열의 합

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

>>> np.sum(A)

15

>>> A.sum()

15

배열의 axis 에 따른 합

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

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

>>> A

array([[0, 1, 2],

       [3, 4, 5]])

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

>>> A.sum(axis=0)

array([3, 5, 7])

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

>>> A.sum(axis=1)

array([ 3, 12])