28. NumPy 어레이 요소 바꾸기¶

numpy.where() 함수를 이용해서 특정 조건에 맞는 어레이의 요소를 대체할 수 있습니다.

특정 값을 다른 값으로 바꾸기¶

import numpy as np

a = np.array([[1, 2], [3, 1]])
b = np.where(a == 1, 10, a)

print(a)
print(b)

[[1 2]
 [3 1]]
[[10  2]
 [ 3 10]]

어레이 a의 요소가 1이라면(a == 1), 값을 10으로 대체합니다.

import numpy as np

a = np.array([[0.5, 1.2, 0.9], [1.1, 0.8, 1.4]])
b = np.where(a < 1.0, 0.0, a)

print(a)
print(b)

[[0.5 1.2 0.9]
 [1.1 0.8 1.4]]
[[0.  1.2 0. ]
 [1.1 0.  1.4]]

어레이 a의 요소가 1.0보다 작으면, 값을 0.0으로 대체합니다.