在NumPy中,可以使用布尔索引来进行条件向量化计算,而无需直接使用掩码。下面是一个示例:
import numpy as np
# 创建一个示例数组
arr = np.array([1, 2, 3, 4, 5])
# 使用布尔索引进行条件向量化计算
result = arr[arr > 2]
print(result)
输出结果为:[3, 4, 5]
在上面的代码中,我们使用布尔索引 arr > 2
来创建一个布尔数组,其中 True
表示满足条件 arr > 2
的元素,而 False
表示不满足条件的元素。然后,我们将布尔数组作为索引应用于原始数组 arr
,以获取满足条件的元素。这样就实现了条件向量化计算,而无需使用直接掩码。