numpy.where¶

numpy.where 함수는

첫번째 인자인 조건식만 주어질 때, 조건에 맞는 요소의 인덱스를 반환합니다.
조건식과 함께 두세번째 인자로 어레이가 함께 주어질 때, 조건식이 참인 경우 두번째 인자의 요소를, 조건식이 거짓인 경우 세번째 인자의 요소를 반환합니다.

예제1 - 조건식만 주어질 때¶

import numpy as np

a = np.array([2, 5, 1, 3, 0])
b = np.where(a < 3)

print(b)
print(a[b])

(array([0, 2, 4], dtype=int64),)
[2 1 0]

np.where(a < 3)는 3보다 작은 어레이 a의 요소의 인덱스를 어레이 형태로 반환합니다.

요소를 출력하려면, a[np.where(a < 3)]의 형태로 사용하면 됩니다.

import numpy as np

a = np.arange(5)
print(a)

b = np.where(a < 3, a, a**2)
print(b)

[0 1 2 3 4]
[ 0  1  2  9 16]

np.where(a < 3, a, a**2)는 어레이 a의 요소가 3보다 작은 경우 어레이 a의 요소를,

그렇지 않은 경우 어레이 a의 요소를 제곱한 값을 반환합니다.