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]