* np.where()
- NumPy 라이브러리에서 제공되는 조건에 따라 배열의 요소를 선택적으로 변경하거나 조건을 만족하는 요소의 인덱스를 반환하는 함수.
- 이 함수는 특정 조건을 검사하고, 해당 조건을 만족하는 요소를 선택적으로 처리하는데 유용.
import numpy as np
# np.where() 함수의 예시
arr = np.array([1, 5, 2, 7, 4, 9])
# 조건을 만족하는 요소의 인덱스를 반환
indices = np.where(arr > 4)
print(indices) # 출력: (array([1, 3, 5]),)
# 조건을 만족하는 요소를 다른 값으로 변경
new_arr = np.where(arr > 4, 0, arr) # 4보다 크면 0으로 변경하고 아닌 값은 그대로.
print(new_arr) # 출력: [1 0 2 0 4 0]
# arr 배열에서 4보다 큰 요소는 0으로 변경, 그렇지 않은 요소는 -1로 변경
new_arr = np.where(arr > 4, 0, -1)
print(new_arr) # 출력: [-1 0 -1 0 -1 0]