np.where函数是什么怎么用
时间: 2024-02-28 13:28:02 浏览: 21
np.where函数是numpy中的一个函数,它用于根据给定条件返回输入数组中的元素的索引。
np.where(condition, [x, y])
参数说明:
condition:条件数组,即需要判断的条件。
x,y:可选参数,用于指定满足条件时输出的值,其中x为满足条件时输出的值,y为不满足条件时输出的值。如果不指定x和y,则返回满足条件的元素的索引。
使用示例:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
# 返回满足条件的元素的索引
print(np.where(arr > 2))
# 返回满足条件的元素的值
print(np.where(arr > 2, arr, 0))
```
输出结果为:
```
(array([2, 3, 4], dtype=int64),)
[0 0 3 4 5]
```
其中,第一个示例返回的是满足条件的元素的索引;第二个示例返回的是满足条件的元素的值,不满足条件则返回0。
相关问题
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,其他元素不变。
如何使用np.where函数进行条件判断?
np.where函数是NumPy库中的一个非常有用的函数,它可以根据给定的条件返回一个新的数组,其中满足条件的元素被替换为指定的值,不满足条件的元素保持不变。下面是使用np.where函数进行条件判断的方法:
1. 导入NumPy库:
```python
import numpy as np
```
2. 创建一个数组:
```python
arr = np.array([1, 2, 3, 4, 5])
```
3. 使用np.where函数进行条件判断:
```python
new_arr = np.where(arr > 3, 0, arr)
```
上述代码中,arr > 3是条件判断语句,如果满足条件,则将对应位置的元素替换为0,否则保持不变。new_arr就是根据条件判断得到的新数组。
注意:np.where函数的第一个参数是条件判断语句,第二个参数是满足条件时替换的值,第三个参数是不满足条件时替换的值。
4. 打印结果:
```python
print(new_arr)
```
输出结果为:
```
[1 2 3 0 0]
```