Python中省掩膜是指對(duì)數(shù)組中的某些元素進(jìn)行操作時(shí),只對(duì)符合條件的元素進(jìn)行操作,不符合條件的元素保持不變的操作方式。下面是一個(gè)例子:
import numpy as np a = np.array([1, 2, 3, 4, 5]) mask = a >3 # 定義省掩膜條件 a[mask] = 0 # 對(duì)符合條件的元素賦值為0 print(a) # 輸出:[1 2 3 0 0]
上面的代碼中,通過(guò)使用`>`符號(hào)定義了一個(gè)省掩膜條件,只對(duì)大于3的元素進(jìn)行操作,將這些元素賦值為0,并輸出修改后的數(shù)組。
Python的NumPy模塊提供了豐富的省掩膜操作,可以通過(guò)各種條件比較、邏輯運(yùn)算符等完成多種元素操作,例如:
import numpy as np a = np.array([1, 2, 3, 4, 5]) mask = (a >2) & (a< 5) # 定義省掩膜條件 a[mask] = a[mask] ** 2 # 對(duì)符合條件的元素賦值為其平方 print(a) # 輸出:[1 2 9 16 5]
上面的代碼中,即使是多個(gè)條件的組合也可以作為省掩膜條件,同時(shí)對(duì)符合條件的元素進(jìn)行平方操作。