[NumPy] np.where()

* 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]