np.where()函数
时间: 2023-07-11 13:53:24 浏览: 40
`np.where()`是numpy库中的一个函数,它的作用是根据条件返回给定数组中符合条件的元素的下标或值。
`np.where()`函数的语法如下:
```python
numpy.where(condition[, x, y])
```
其中,参数说明如下:
- `condition`:条件,可以是bool型数组或bool型表达式。
- `x`:满足条件(condition)时返回的数组,可以是numpy数组或标量。
- `y`:不满足条件(condition)时返回的数组,可以是numpy数组或标量。如果没有指定,则返回满足条件 (condition) 的元素的索引。
`np.where()`函数返回一个元组,包含符合条件的元素的下标或值。
例如,对于一个numpy数组`arr`,我们可以使用`np.where()`函数来查找其中所有大于0的元素的下标:
```python
import numpy as np
arr = np.array([-2, 0, 3, -1, 5, -4, 6])
indexes = np.where(arr > 0)
print(indexes)
```
输出结果为:
```
(array([2, 4, 6], dtype=int64),)
```
其中,`(array([2, 4, 6], dtype=int64),)`表示符合条件的元素在原数组中的下标为2、4、6。
相关问题
np.where函数
np.where函数是numpy库中的一个非常实用的函数,主要有两个用法。第一个用法是np.where(condition, x, y),它会返回满足condition的索引。第二个用法是首先传入y,然后如果满足条件condition,那么y中原来的数值变为x,否则不变化。相当于[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]。这个函数可以帮助我们快速地对数组进行筛选和替换操作。
python中np.where函数
np.where函数是numpy库中的一个函数,它的作用是根据指定的条件,返回符合条件的元素的索引或值。np.where函数的语法如下:
np.where(condition[, x, y])
其中,condition是指定的条件,x和y是可选的参数,用于指定符合条件和不符合条件的返回值。
当只有condition参数时,np.where函数返回符合条件的元素的索引。
当有x和y参数时,np.where函数会根据condition的值,返回符合条件的元素的值或不符合条件的元素的值。
例如,以下代码会将数组a中小于0的元素替换为0,其他元素不变:
```python
import numpy as np
a = np.array([-1, 2, -3, 4, 5])
b = np.where(a < 0, 0, a)
print(b)
```
输出结果为:
```
[0 2 0 4 5]
```
其中,np.where(a < 0, 0, a)的作用是将数组a中小于0的元素替换为0,其他元素不变。