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]])