[파이썬] array 배열에서 최솟값과 최댓값의 인덱스 반환하는 방법

 

array 배열에서 최솟값과 최댓값을 구하려면 np.min(), np.max() 함수를 사용하면 되는데요,

그렇다면 최솟값과 최댓값의 인덱스 번호를 알고 싶다면 어떻게 해야할까요?

최솟값과 최댓값의 인덱스를 반환해주는 함수가 바로 argmin(), argmax() 함수인데요, 아래 코드를 보며 설명하도록 하겠습니다.

 

 

- argmin(), argmax() : 최솟값, 최댓값의 인덱스 반환

argmin(), argmax() 함수는 array에서 최솟값과 최댓값의 인덱스를 반환해주는 함수인데요,

먼저, 아래와 같은 array가 있다고 가정해보겠습니다.

 

import numpy as np
arr1 = np.random.randn(2,5)
print(arr1)

 

[[ 2.69686599  0.19144308  1.46021652 -0.87155883 -1.18851542]
 [-0.69282548 -1.5482184   1.18121393  1.38972697 -0.88128495]]

 

 

그리고나서, argmin, argmax 함수를 통해 해당 인덱스를 반환해보도록 하겠습니다.

 

print(np.argmin(arr1))
print(np.argmax(arr1))

 

6
0

 

출력된 값을 보면, 최솟값의 인덱스번호는 6, 최댓값의 인덱스번호는 0인데요, 위 array의 최솟값과 최댓값의 인덱스가 잘 반환되었음을 확인할 수 있습니다.