Question

假设我有一个 numpy ndarray

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

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

[[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 ndarry 负元素

2 个答案: