[Numpy] where 함수 사용법

numpy의 where 함수를 사용하는 방법에 대해서 알아보자.

---

numpy의 where 함수는 사용 방법을 크게 두가지로 나눌 수 있다.

 

1. 조건을 만족하는 인덱스 값 가져오기

test_a = np.arange(10)

print(np.where(test_a > 5))
# 출력: (array([6, 7, 8, 9]),)

 

2. True일 때, False일 때 각각 값을 broadcasting

조건문에서 True가 반환되는 위치와 False가 반환되는 위치에 각각 정해진 값을 넣는다.

np.where(조건문, True일 때 값, False일 때 값)

test_a = np.arange(10)

print(np.where(test_a > 5, 1, -1))
# 출력: [-1 -1 -1 -1 -1 -1  1  1  1  1]