NumPy使用技巧1：条件运算

在NumPy中，条件运算允许你根据条件对数组的元素进行逐个操作，无需显式地编写循环

将条件应用于数组，从而创建一个布尔数组，其中的元素表示原数组每个位置是否满足条件

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = arr > 3 # 创建布尔数组
print(condition)

输出结果为：

[False False False True True]

可用于选出原数组中满足条件的元素

selection = arr[condition] # 选择满足条件的元素
print(selection)

输出结果为：

[4 5]

使用布尔数组，将原数组中满足条件的元素置为特定值

arr[condition] = 0 # 将大于3的元素置为0
print(arr)

输出结果为：

[1 2 3 0 0]

可以使用逻辑运算符&（与），|（或），~（非）来组合多个布尔数组的条件

condition_combined = (arr > 1) & (arr < 4)
selection_combined = arr[condition_combined]

输出结果为：

[2 3]