解释一下np.where的参数及用法
时间: 2023-10-19 12:03:24 浏览: 76
Python astype(np.float)函数使用方法解析
np.where函数的参数非常简单明了,可以分为三个参数,分别是(condition, x, y)。
其中condition是一个bool型的数组,x和y分别是两个同型数组,用于根据condition的True或False,分别从x和y中筛选相应的元素构成输出数组。
举个例子,如果condition数组中某个位置为True,则输出数组中对应位置的元素来自x数组; 如果condition数组中某个位置为False,则输出数组中对应位置的元素来自y数组。
例如,如下代码可以实现对array数组求正数部分的功能:
```python
import numpy as np
array = np.array([-1, 3, -5, 7, -9])
positive = np.where(array > 0, array, 0)
print(positive)
# 输出 [0 3 0 7 0]
```
在这个例子中,我们将condition数组设为array > 0,x数组设为array,y数组设为0。当array中的某个元素大于0时,输出数组中对应位置的元素来自array,否则为0。
希望这个解释能帮到您。
阅读全文