用 0（零）替换 numpy ndarray 负元素

Question

假设我有一个 numpy ndarray

[[2, -7, 5], [-6, 2, 0] [1, -4, 2], [-2, 6, 8]]

如何得到一个所有负元素都被0替换的numpy nd arry:

[[2, 0, 5], [0, 2, 0] [1, 0, 2], [0, 6, 8]]

感谢您抽出宝贵时间。

Answer 1

您可以为此使用 np.clip，在零和无穷大之间剪裁：

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

np.clip(arr, a_min = 0, a_max = np.inf)

array([[2., 0., 5.],
       [0., 2., 0.],
       [1., 0., 2.],
       [0., 6., 8.]])

否则，你可以使用这样的东西（注意这会改变数组）：

arr[arr <= 0] = 0

>>> arr
array([[2, 0, 5],
       [0, 2, 0],
       [1, 0, 2],
       [0, 6, 8]])

Answer 2

您可以使用 np.where() 来：

arr = np.array([[2, -7, 5], [-6, 2, 0], [1, -4, 2], [-2, 6, 8]])
result = np.where(arr<0, 0, arr)

输出：

[[2 0 5]

 [0 2 0]
 [1 0 2]
 [0 6 8]]

用 0（零）替换 numpy ndarray 负元素

Replace numpy ndarry negative elements with 0 (zero)

python

numpy

numpy-ndarray