如何在 numpy 中用零替换负值

您可以使用以下基本语法在 NumPy 中用零替换负值：

 my_array[my_array < 0 ] = 0

此语法适用于 1D 和 2D NumPy 数组。

以下示例展示了如何在实践中使用此语法。

示例 1：在一维 NumPy 数组中用零替换负值

以下代码展示了如何将 NumPy 数组中的所有负值替换为零：

 import numpy as np

#create 1D NumPy array
my_array = np. array ([4, -1, 6, -3, 10, 11, -14, 19, 0])

#replace negative values with zero in array
my_array[my_array < 0 ] = 0

#view updated array
print (my_array)

[4 0 6 0 10 11 0 19 0]

请注意，原始数组中的每个负值都已替换为零。

示例 2：在 2D NumPy 数组中用零替换负值

假设我们有以下 2D NumPy 数组：

 import numpy as np

#create 2D NumPy array
my_array = np. array ([3, -5, 6, 7, -1, 0, -5, 9, 4, 3, -5, 1]). reshape (4.3)

#view 2D NumPy array
print (my_array)

[[ 3 -5 6]
 [ 7 -1 0]
 [-5 9 4]
 [3 -5 1]]

我们可以使用以下代码将 NumPy 数组中的所有负值替换为零：

 #replace all negative values with zero in 2D array
my_array[my_array < 0 ] = 0

#view updated array
print (my_array)

[[3 0 6]
 [7 0 0]
 [0 9 4]
 [3 0 1]]

请注意，原始二维数组中的所有负值均已替换为零。

其他资源

以下教程解释了如何在 NumPy 中执行其他常见任务：

如何用值填充 NumPy 数组
如何从 NumPy 数组中删除特定元素
如何替换 NumPy 数组中的元素
如何从 NumPy 数组中获取特定行