Python에서 numpy의 범위 내 요소 찾기

numpy 라이브러리를 사용하여 데이터를 처리하는 동안 특정 범위의 특정 숫자를 필터링해야 할 수도 있습니다. 이것은 numpy에서 사용할 수 있는 몇 가지 내장 메서드를 사용하여 달성할 수 있습니다.

및 연산자

이 접근 방식에서 우리는 numpy 배열을 취하고 여기에 logical_and 함수를 적용합니다. numpy의 where 절은 and 조건을 적용하는 데에도 사용됩니다. 결과는 필요한 범위 조건을 충족하는 요소의 위치를 ​​​​나타내는 배열입니다.

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = np.where(np.logical_and(A >= 6, A <= 15))

# Result
print("Array with condition : ", res)

출력

위의 코드를 실행하면 다음과 같은 결과가 나옵니다. -

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : (array([1, 2, 6], dtype=int32),)

* 사용

이 접근 방식에서는 * 연산자를 사용합니다. 배열에서 값의 위치가 아닌 실제 값으로 결과를 제공합니다.

import numpy as np

A = np.array([5, 9, 11, 4, 31, 27,8])

# printing initial array
print("Given Array : ", A)

# Range 6 to 15
res = A [ (A >=6) * (A <= 15)]

# Result
print("Array with condition : ", res)

출력

위의 코드를 실행하면 다음과 같은 결과가 나옵니다. -

Given Array : [ 5 9 11 4 31 27 8]
Array with condition : [ 9 11 8]