[파이썬 강의] 73. 배열 원소의 합

배열의 합을 구할 때는 sum 매소드를 사용합니다. 2차원 이상의 배열에서는 axis 옵션을 사용할 수 있습니다. 

 

1차원 배열

1차원 배열을 하나 정의하고 합을 구해봅시다. 

>>> import numpy as np
>>> v1=np.array([1,2,3,4,5])
>>> v1.sum()
15

 

2차원 배열

2차원 배열을 하나 정의하고 합을 구해봅시다. 

>>> m1=np.array([[1,2,3],[10,20,30]])
>>> m1.sum()
66

axis 옵션을 적용해봅시다. 

>>> m1.sum(axis=0)
array([11, 22, 33])
>>> m1.sum(axis=1)
array([ 6, 60])

axis0은 [1,2,3]->[10,20,30] 이므로 리스트 [1,2,3]과 [10,20,30]의 합입니다. 

axis1은 1->2->3 과 10->20->30 이므로 1,2,3의 합과 10,20,30의 합입니다.