[Numpy] 조건을 만족하는 위치의 인덱스 찾기 (np.where)

[Numpy] 조건을 만족하는 위치의 인덱스 찾기 (np.where)

pandas의 loc처럼 numpy의 np.where() 함수를 사용하면 배열 내의 특정 조건을 만족하는

원소의 index를 찾아낼 수 있다. 

 

만약 배열 내에서 짝수인 요소의 인덱스를 찾으려면 다음과 같이 할 수 있다.

import numpy as np

arr = np.array([2,3,6,7,10,13,14])
np.where(arr%2==0)

 

loc처럼 특정 조건을 만족했을 때, 대체하는 것 또한 가능하다.

요소가 짝수면 2를 반환하고 홀수면 1을 반환해 보자

arr2 = np.where(arr%2==0, 2, 1)

 

다음과 같이 조건에 맞는 인덱스를 찾아 수정할 수 있다.

 

참고

https://jimmy-ai.tistory.com/46

https://studyweb.tistory.com/entry/Numpy-%EC%82%AC%EC%9A%A9%EB%B2%95-5-npwhere