NumPy는 논리 연산자를 사용하여 배열 요소에 대한 조건을 정의합니다. 먼저, 다음과 같은 NumPy 배열을 가정해보겠습니다.

import numpy as np

arr = np.array([1, 2, 3, 4, 5])

특정 조건을 만족하는 배열 요소를 검색하려면 다음과 같이 작성할 수 있습니다.

# 조건을 만족하는 요소의 인덱스를 가져옵니다.
indices = np.where(arr > 3)

# 인덱스를 사용하여 조건을 만족하는 요소를 가져옵니다.
result = arr[indices]

print(result)  # [4, 5]

위의 코드에서 np.where() 함수는 조건을 만족하는 요소의 인덱스를 반환합니다. 이러한 인덱스를 사용하여 배열에서 실제 값을 가져올 수 있습니다.

NumPy에서는 비교 연산자 외에도 논리 연산자를 사용하여 여러 조건을 결합할 수도 있습니다. 예를 들어, 다음과 같이 두 가지 조건을 동시에 만족하는 요소를 검색할 수 있습니다.

indices = np.where((arr > 2) & (arr < 5))

더 복잡한 조건을 위해 논리 연산자 &와 |를 사용하여 조건을 결합할 수도 있습니다.

NumPy 배열에서 특정 조건을 만족하는 요소를 검색하는 방법을 소개했습니다. 더 자세한 내용은 NumPy 공식 문서를 참조하시기 바랍니다.

• NumPy 공식 문서: https://numpy.org/doc/