Python NumPy where方法用法及代码示例

例子

基本测绘

要将所有 True 替换为 5，将所有 False 替换为 10：

x = np.array([True, True, False, False])
np.where(x, 5, 10)



array([5, 5, 10, 10])

使用布尔掩码

更实际的是，where() 方法与布尔掩码配合使用效果很好。假设我们希望所有大于 2 的值均为 10，所有其他值均为 -1。

我们首先构建一个布尔掩码，如下所示：

x = np.array([1,2,3,4])
mask = x > 2
mask



array([False, False, True, True])

然后我们应用映射：

np.where(mask, 10, -1)



array([-1, -1, 10, 10])

根据数字替换值

假设我们想根据特定标准执行一些算术。例如，我们要对大于 2 的值加 5，否则减 5：

x = np.array([1,2,3,4])
np.where(x>2, x+5, x-5)



array([-4, -3, 8, 9])

多维数组

方法where(~)也可用于多维数组：

x = np.array([[1,2],[3,4]])
np.where(x>2, 10, 5)



array([[5, 5],
       [10, 10]])