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)]의 형태로 사용하면 됩니다.



예제2 - 세 개의 인자가 모두 주어질 때

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의 요소를 제곱한 값을 반환합니다.


관련 페이지


이전글/다음글

이전글 :
다음글 :