(파이썬) numpy.amax

numpy 모듈의 amax 함수는 array 의 최댓값을 반환하는 함수이다. 다음 예를 보자.

>>> import numpy as np
>>> a = np.range(4)
>>> a
array([0, 1, 2, 3])
>>> np.amax(a)
3

위 예에서 볼 수 있듯이 np.amax(a) 는 array [0, 1, 2, 3] 의 요소 중 가장 큰 값이 3을 반환하는 거을 알 수 있다.

amax 함수는 행렬 (matrix) 에서 행이나 열 각각에 대한 최댓값을 반환해 주기도 한다.

>>> a = np.arange(4).reshape((2, 2))
>>> a
array([[0, 1],
       [2, 3]])
>>> np.amax(a, axis=0)
array([2, 3])
>>> np.amax(a, axis=1)
array([1, 3])

reshape((2, 2)) 는 array [0, 1, 2, 3] 을 $2 \times 2$ 행렬 $\begin{bmatrix}0&1\\2&3\\ \end{bmatrix}$ 으로 만들어 준다.

np.amax(a, axis=0) 으로 준 경우 각 열의 최댓값로 구성된 array [2, 3] 을 리턴한다. 즉, 1열 $\begin{bmatrix} 0\\2\\ \end{bmatrix}$ 의 최댓값은 2 이고, 2열 $\begin{bmatrix} 1\\3\\ \end{bmatrix}$ 의 최댓값은 3 이므로 array [2, 3] 이 리턴되는 것이다.

np.amax(a, axis=1) 로 경우는 각 행의 최댓값으로 구성된 array [1, 3] 을 리턴한다. 즉, 1행 $\begin{bmatrix} 0&1 \end{bmatrix}$ 의 최댓값은 1이고, 2행 $\begin{bmatrix} 2&3 \end{bmatrix}$ 의 최댓값은 3 이므로 array [1, 3]  이 리턴되는 것이다.