np.where怎么用
时间: 2023-07-08 13:19:45 浏览: 19
np.where() 是 numpy 库中的一个函数,用于根据一个条件从两个数组当中选择元素。
函数的使用格式如下:
```python
np.where(condition, x, y)
```
参数说明:
- condition:一个布尔型数组或条件表达式。
- x:满足条件时的结果。
- y:不满足条件时的结果。
函数的返回值是一个数组,其元素来源于 x 和 y。
下面是一个示例代码:
```python
import numpy as np
# 创建两个数组
a = np.array([0, 1, 2, 3, 4, 5])
b = np.array([10, 20, 30, 40, 50, 60])
# 选择元素
c = np.where(a > 3, b, a)
print(c)
```
输出结果为:
```
[ 0 1 2 3 50 60]
```
在上面的代码中,当 a 数组中的元素大于 3 时,选择 b 数组中的元素,否则选择 a 数组中的元素。
相关问题
torch.where np.where
torch.where()是PyTorch和NumPy中的函数,用于根据给定的条件选择元素。torch.where()用于PyTorch张量,而np.where()用于NumPy数组。
在PyTorch中,torch.where()函数接受一个条件和两个张量作为输入。它会根据条件选择元素,返回与条件为True的元素对应的索引。例如,如果我们有一个张量x,我们可以使用torch.where(x==0)来找到x中为0的元素的索引。
在NumPy中,np.where()函数的功能与torch.where()类似,但输入和输出的类型不同。np.where()函数接受一个条件和一个数组作为输入,并返回与条件为True的元素对应的索引。例如,如果我们有一个数组arr,我们可以使用np.where(arr==0)来找到arr中为0的元素的索引。
np.where的使用
np.where()函数可以根据条件返回数组中符合条件的元素的索引或值。它的语法如下:
np.where(condition, [x, y])
其中,condition是一个布尔数组,x和y是可选参数。如果只有condition参数,则返回一个元素为True的索引数组;如果同时有x和y参数,则返回一个数组,其元素的值根据condition数组的值来决定,即当condition为True时,返回x的对应元素,否则返回y的对应元素。
例如,下面的代码演示了如何使用np.where()函数:
```python
import numpy as np
# 创建一个数组
arr = np.array([1, 2, 3, 4, 5])
# 获取所有大于3的元素的索引
indices = np.where(arr > 3)
print(indices)
# 根据条件返回不同的值
new_arr = np.where(arr > 3, arr, 0)
print(new_arr)
```
输出结果为:
```
(array([3, 4]),)
[0 0 0 4 5]
```
其中,`np.where(arr > 3)`返回了一个元素为True的索引数组,即[3, 4];而`np.where(arr > 3, arr, 0)`返回了一个新的数组,其中大于3的元素的值保持不变,小于等于3的元素的值被替换为0。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)