[Python] Numpy argmax( ),argmin( ),np.where( ) - 최소, 최대, 조건 색인값 찾기

[Python] Numpy argmax( ),argmin( ),np.where( ) - 최소, 최대, 조건 색인값 찾기

 

 

 

np.min(), np.max()

 

>>> import numpy as np

>>> x = [5,6,8,1,3,9,10]
>>> np.min(x)

1

np.min()은 리스트안에 최소값을 찾을 수 있다.

 

>>> np.max(x)

10

np.max()는 리스트안에 최대값을 찾을 수 있다.

 

 

 

 

np.argmin(), np.argmax()

 

최소값, 최대값 색인 위치 찾기.

쉽게 말하면 최대값이 몇번째에 있는지 확인 할 수 있는 코드이다.

>>> x = [5,6,8,1,3,9,10]
>>> np.argmin(x)

3

 

리스트 x의 최소값은 1이다.

1의 위치는 (0,1,2,3) 3번째이다. - 파이썬에서 리스트는  0부터 시작한다.

 

>>> np.argmax(x)

6

리스트 x의 최대값 10의 위치는 6번째이다.

 

 

 

np.where()

 

조건을 넣어서 활용할 수 있다.

 

위의 np.max / np.min 과 활용하는 법이 조금 다르다.

>>> import numpy as np
>>> a = np.array([0,5,8,10])
>>> b = np.where(a > 5)
>>> print(b)

(array([2, 3]),)

a에 np.array로 인덱스를 만들어 할당시키고

b에 np.where로 조건을 주어 할당시킨후

프린트한다.

 

a가 5보다 크다는 조건을 달았기 때문에

5보다 큰 8과 10의 순서 2와 3이 출력된다.