여러 조건으로 numpy where()를 사용하는 방법

다음 방법을 사용하여 여러 조건으로 NumPywhere() 함수를 사용할 수 있습니다.

방법 1: OR과 함께 Where() 사용

 #select values less than five or greater than 20
x[np. where ((x < 5) | (x > 20))]

방법 2: AND와 함께 Where() 사용

 #select values greater than five and less than 20
x[np. where ((x > 5) & (x < 20))]

다음 예에서는 각 방법을 실제로 사용하는 방법을 보여줍니다.

방법 1: OR과 함께 Where() 사용

다음 코드는 NumPy 배열에서 5보다 작 거나 20보다 큰 각 값을 선택하는 방법을 보여줍니다.

 import numpy as np

#define NumPy array of values
x = np. array ([1, 3, 3, 6, 7, 9, 12, 13, 15, 18, 20, 22])

#select values that meet one of two conditions
x[np. where ((x < 5) | (x > 20))]

array([ 1, 3, 3, 22])

NumPy 배열의 4개 값은 5보다 작 거나 20보다 컸습니다.

또한 크기 함수를 사용하여 다음 조건 중 하나를 충족하는 값의 수를 간단히 찾을 수도 있습니다.

 #find number of values that are less than 5 or greater than 20
(x[np. where ((x < 5) | (x > 20))]). size

4

방법 2: AND와 함께 Where() 사용

다음 코드는 5보다 크고 20보다 작은 NumPy 배열에서 각 값을 선택하는 방법을 보여줍니다.

 import numpy as np

#define NumPy array of values
x = np. array ([1, 3, 3, 6, 7, 9, 12, 13, 15, 18, 20, 22])

#select values that meet two conditions
x[np. where ((x > 5) & (x < 20))]

array([6, 7, 9, 12, 13, 15, 18])

출력 배열에는 5 보다 크고 20보다 작은 원래 NumPy 배열의 7개 값이 표시됩니다.

이번에도 크기 함수를 사용하여 두 조건을 모두 충족하는 값의 수를 확인할 수 있습니다.

 #find number of values that are greater than 5 and less than 20
(x[np. where ((x > 5) & (x < 20))]). size

7

추가 리소스

다음 튜토리얼에서는 NumPy에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

NumPy 배열의 모드를 계산하는 방법
NumPy 배열에서 값 인덱스를 찾는 방법
NumPy 배열에 함수를 매핑하는 방법