[Python] 넘파이 조건에 부합하는 데이터/인덱스 위치 찾기: argmax(), argmin()

넘파이 안에 조건에 부합하는 데이터 찾기

 

ex) X에 70보다 큰 데이터는 몇개가 있나요?

- 컴퓨터에서는 True를 1로, False를 0으로 계산하기 때문에, 총합을 계산하면, 총 개수가 나온다. 

- (X>70).sum()

 

ex) X의 데이터 중에서 70보다 큰 데이터만 가져오시오.

 

최대값/ 최소값 인덱스 위치 찾기

 

- ex) 최대값은 어디에 있나요? = 최대값이 들어있는 곳의 인덱스가 무엇? 

- 넘파이 변수. argmax()

 

 

- 같은 원리로, 최소값이 들어있는 곳의 인덱스가 무엇? 

- 넘파이 변수. argmin()

 

ex) 각 행별로 최대값의 인덱스를 구하시오.

X.argmax(1)

 

ex) 각 열별로 최소값의 인덱스를 구하시오

X.argmin(axis=0)