np.where 输入多个条件
时间: 2024-05-27 10:05:32 浏览: 97
python numpy之np.random的随机数函数使用介绍
5星 · 资源好评率100%
np.where函数可以根据多个条件来选择元素。它的语法如下:
np.where(条件, 条件成立时的返回值, 条件不成立时的返回值)
其中,条件可以是一个布尔数组或者一个布尔表达式,用于指定元素是否满足条件。条件成立时,返回值为条件成立时的返回值;条件不成立时,返回值为条件不成立时的返回值。
如果要输入多个条件,可以使用逻辑运算符(如and、or)将多个条件组合起来。例如,要选择满足两个条件的元素,可以使用逻辑与运算符(&)将两个条件连接起来。
下面是一个示例:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition1 = arr > 2
condition2 = arr < 5
result = np.where(condition1 & condition2, arr, 0)
print(result)
```
输出结果为:[0 0 3 4 0]
在这个示例中,我们定义了两个条件:condition1表示元素大于2,condition2表示元素小于5。然后使用逻辑与运算符(&)将两个条件连接起来,得到一个新的条件。最后,使用np.where函数根据新的条件选择元素,满足条件的元素保持原值,不满足条件的元素被替换为0。
阅读全文