[파이썬][Numpy] 배열의 비교 및 불리언 연산 예제

24 Aug 2023

numpy

Numpy를 사용하여 배열의 비교와 불리언(Boolean) 연산을 수행하는 예제를 제공하겠습니다. 배열 간의 비교 및 불리언 연산은 데이터 필터링, 조건부 연산, 마스크(mask) 생성 등 다양한 작업에서 유용하게 활용됩니다.

import numpy as np

# 두 개의 배열 생성
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([4, 5, 6, 7, 8])

# 배열 비교 연산
comparison = arr1 == arr2
print("두 배열의 비교 결과:")
print(comparison)

# 불리언 배열 생성
boolean_array = arr1 > 3
print("\narr1에서 3보다 큰 요소를 가진 불리언 배열:")
print(boolean_array)

# 불리언 배열을 사용한 데이터 필터링
filtered_array = arr1[boolean_array]
print("\n3보다 큰 요소를 가진 arr1의 필터링된 배열:")
print(filtered_array)

# 복합 불리언 연산
combined_boolean = (arr1 > 2) & (arr2 < 7)
print("\narr1에서 2보다 크고, arr2에서 7보다 작은 요소를 가진 복합 불리언 배열:")
print(combined_boolean)

위의 코드에서는 두 개의 배열을 생성하고, 배열 간의 비교 연산을 수행하여 불리언(Boolean) 배열을 생성합니다. 이후 불리언 배열을 사용하여 데이터를 필터링하거나 복합 불리언 연산을 수행하는 예제를 보여줍니다. 이러한 불리언 연산을 사용하면 배열에서 조건을 충족하는 요소를 선택하거나 데이터를 처리하는 데 유용합니다.