Numpy array 음수값 대체하기(how to replace negative values in numpy array)

안녕하세요 jay입니다.
오늘은 numpy array에서 음수값을 특정 값으로 대체하는 방법에 대해
알아보도록 하겠습니다.

# replace negative values with positive values

import numpy as np

# make identity matrix(3x3)
arr = np.eye(3)
print(arr)

"""
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
"""

# replace 1 with -1
arr[arr > 0] = -1

print(arr)
"""
[[-1. -0. -0.]
 [-0. -1. -0.]
 [-0. -0. -1.]]
"""

# replace negative values with 0
arr[arr < 0] = 0
print(arr)

먼저 예시를 위해 단위행렬인 numpy array를 만들었습니다. 그 후 1을 가지면 -1로 대체를 했습니다. 그리고 음수일 때, 0으로 대체를 했습니다. 위와 같이 하는 법도 있지만 np.where를 쓰면 더 다채롭게 변형이 가능합니다. 

링크 : https://classicismist.blogspot.com/2018/10/numpy-array-npwhere.html