python省掩膜

Python中省掩膜是指对数组中的某些元素进行操作时，只对符合条件的元素进行操作，不符合条件的元素保持不变的操作方式。下面是一个例子：

import numpy as np
a = np.array([1, 2, 3, 4, 5])
mask = a >3  # 定义省掩膜条件
a[mask] = 0  # 对符合条件的元素赋值为0
print(a)  # 输出：[1 2 3 0 0]

上面的代码中，通过使用`>`符号定义了一个省掩膜条件，只对大于3的元素进行操作，将这些元素赋值为0，并输出修改后的数组。

Python的NumPy模块提供了丰富的省掩膜操作，可以通过各种条件比较、逻辑运算符等完成多种元素操作，例如：

import numpy as np
a = np.array([1, 2, 3, 4, 5])
mask = (a >2) & (a< 5)  # 定义省掩膜条件
a[mask] = a[mask] ** 2  # 对符合条件的元素赋值为其平方
print(a)  # 输出：[1 2 9 16 5]

上面的代码中，即使是多个条件的组合也可以作为省掩膜条件，同时对符合条件的元素进行平方操作。