[Python] Numpy 최댓값, 최솟값의 인덱스 구하기

아래와 같은 넘파이 어레이가 있을때 , 

최댓값 인덱스와, 최솟값 인덱스를 알아보자

>>> X
array([[36, 66, 62, 17, 50],
       [58, 47,  6, 45, 72],
       [11,  1,  2, 85, 97],
       [12, 64, 21,  3, 33]])

최댓값이 들어있는 곳의 인덱스 찾기

>>> X.argmax()
14

 

최솟값이 들어있는 곳의 인덱스 찾기

>>> X.argmin()
11

 

각 행별로 최댓값의 인덱스 구하기

>>> X.argmax(axis = 1)
array([1, 4, 4, 1], dtype=int64)

 

각 열별로 최댓값의 인덱스 구하기

>>> X.argmin(axis = 0)
array([2, 2, 2, 3, 3], dtype=int64)