Python NumPy putmask方法用法及代碼示例

Numpy 的 putmask(~) 在給定布爾掩碼的輸入數組中設置特定值。這是就地完成的，即不會創建新的 Numpy 數組。

1. a | array-like

輸入數組。所有數組都被視為展平數組。

2. mask | array-like 共 booleans

指示是否設置該值的布爾值數組。形狀必須與 a 相同。

3. values | array-like

要設置的值。如果 v 比 a 短，則將重複 v 以確保形狀匹配。

None - 設置就地完成。

a = np.array([5,6,7,8])
np.putmask(a, [True, False, True, False], 9)
a



array([9, 6, 9, 8])

請注意如何僅設置掩碼中具有相應 True 值的值。

考慮以下二維數組：

a = np.array([[5,6],[7,8]])
a



array([[5, 6],
       [7, 8]])

np.putmask(a, [True, False, True, False], 9)
a



array([[9, 6],
       [9, 8]])

所有數組都被視為展平數組。

a = np.array([[5,6],[7,8]])
np.putmask(a, [True, True, False, False], [9,10])
a



array([[ 9, 10],
       [ 7,  8]])

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | putmask method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。