np.where的参数
时间: 2024-02-16 15:58:00 浏览: 54
Python astype(np.float)函数使用方法解析
np.where()函数是一个条件判断函数,用于根据条件返回数组中的元素。它的参数包括三个:
1. condition:一个布尔数组或条件表达式,用于指定条件。
2. x:一个数组或标量,用于指定满足条件的元素的替代值。
3. y:一个数组或标量,用于指定不满足条件的元素的替代值。
下面是一个示例代码:
```python
import numpy as np
x = np.array([1, 2, 3, 4, 5])
y = np.array([10, 20, 30, 40, 50])
result = np.where(x > 3, x, y)
print(result)
```
输出结果为:
```
[10 20 30 4 5]
```
在这个示例中,我们使用np.where()函数根据条件x > 3来选择元素。如果满足条件,就选择x中的元素,否则选择y中的元素。
阅读全文