[파이썬][Scipy] 조건부 인덱싱 예제

25 Aug 2023

scipy

조건부 인덱싱은 조건을 충족하는 배열의 요소만 선택하는 데 사용됩니다. NumPy 및 Scipy에서 조건부 인덱싱을 수행하는 예제를 제공합니다.

1차원 배열에서의 조건부 인덱싱 예제:

mport numpy as np

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

# 조건을 만족하는 요소만 선택
condition = arr > 2
filtered_arr = arr[condition]

print("원본 배열:")
print(arr)
print("조건부 인덱싱을 통해 선택된 요소:")
print(filtered_arr)

2차원 배열에서의 조건부 인덱싱 예제:

mport numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 조건을 만족하는 요소만 선택
condition = arr > 4
filtered_arr = arr[condition]

print("원본 배열:")
print(arr)
print("조건부 인덱싱을 통해 선택된 요소:")
print(filtered_arr)

부울 배열을 사용한 조건부 인덱싱 예제:

mport numpy as np

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

# 부울 배열을 생성하여 조건을 만족하는 요소를 선택
bool_indices = [True, False, True, False, True]
filtered_arr = arr[bool_indices]

print("원본 배열:")
print(arr)
print("부울 배열을 사용한 조건부 인덱싱 결과:")
print(filtered_arr)

조건부 인덱싱은 데이터 필터링과 데이터 분석에서 매우 유용하며, 조건에 따라 배열의 특정 요소를 선택하거나 수정하는 데 사용됩니다.