[Python/파이썬] Numpy - where() 사용 방법

예제 1: 조건에 따라 배열 값 변경

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr < 3, 0, arr)
print(new_arr)

위의 예제는 배열에서 조건에 따라 값을 변경하는 방법을 보여줍니다. 조건에 해당하는 요소는 0으로 변경되고, 그렇지 않은 경우에는 원래의 값이 유지됩니다.

예제 2: 조건에 맞는 요소의 인덱스 반환

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
indices = np.where(arr > 2)
print(indices)

이 예제에서는 배열에서 조건을 만족하는 요소의 인덱스를 반환합니다. 반환되는 값은 해당 요소들의 인덱스를 담은 배열입니다.

예제 3: 두 배열을 비교하여 조건에 따라 값 변경

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([5, 4, 3, 2, 1])
new_arr = np.where(arr1 > arr2, arr1, arr2)
print(new_arr)

이 예제에서는 두 배열을 비교하여 조건에 따라 값을 변경합니다. arr1의 값이 arr2보다 크면 arr1의 값으로 변경되고, 그렇지 않은 경우에는 arr2의 값이 유지됩니다.

예제 4: 조건을 만족하는 요소의 개수 세기

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
count = np.where(arr % 2 == 0)[0].size
print(count)

이 예제에서는 배열에서 조건을 만족하는 요소의 개수를 세는 방법을 보여줍니다. 반환된 인덱스 배열의 크기를 이용하여 개수를 구합니다.

예제 5: 조건을 만족하는 요소에 대한 연산

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
result = np.where(arr % 2 == 0, arr * 2, arr)
print(result)

이 예제에서는 배열에서 조건을 만족하는 요소에 대해 특정 연산을 수행합니다. 짝수인 경우에는 값에 2를 곱하고, 홀수인 경우에는 원래의 값이 유지됩니다.

예제 6: 조건을 만족하지 않는 요소의 인덱스 반환

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
indices = np.where(arr != 3)
print(indices)

이 예제에서는 배열에서 조건을 만족하지 않는 요소의 인덱스를 반환합니다. 반환되는 값은 해당 요소들의 인덱스를 담은 배열입니다.