Programming/Python
[Python] Numpy argmax( ),argmin( ),np.where( ) - 최소, 최대, 조건 색인값 찾기
EunjiBest
2022. 1. 20. 17:46
[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이 출력된다.
반응형