np.argwhere什么意思
时间: 2023-11-17 19:06:50 浏览: 51
`np.argwhere()` 是 NumPy 库中的一个函数,用于返回数组中非零元素的索引。它返回一个数组,其中每个行为非零元素的索引。例如,对于数组 `arr`,`np.argwhere(arr)` 返回的是一个形状为 `(n, arr.ndim)` 的数组,其中 `n` 是数组中非零元素的数量,`arr.ndim` 是数组的维度数。每一行表示一个非零元素的索引,其中每个元素对应该维度的索引值。
相关问题
np.argwhere这个函数什么意思
`np.argwhere` 是 NumPy 库中的一个函数,用于在数组中查找非零元素的索引。它将返回一个数组,其中每个元素是一个非零元素的索引,以元组的形式表示。例如,对于以下数组:
```python
import numpy as np
arr = np.array([[0, 1, 0],
[1, 0, 1],
[0, 1, 0]])
```
调用 `np.argwhere` 函数将返回一个包含非零元素索引的数组:
```python
np.argwhere(arr)
# 输出:array([[0, 1],
# [1, 0],
# [1, 2],
# [2, 1]])
```
这表示第 0 行第 1 列、第 1 行第 0 列、第 1 行第 2 列、第 2 行第 1 列这些位置的元素都是非零元素。
np.argwhere
np.argwhere函数是NumPy库中的一个函数,用于查找数组中满足某个条件的元素的索引。它返回一个包含满足条件的元素索引的数组。具体来说,np.argwhere(condition)会返回一个二维数组,其中每一行是满足条件的元素的索引。条件可以是任何能够返回布尔值的表达式或函数。
例如,假设有一个一维数组arr,我们想要找到数组中所有大于5的元素的索引,可以使用np.argwhere(arr > 5)。
请注意,np.argwhere返回的是多维数组的索引,所以结果是一个二维数组。如果想要得到一维数组形式的索引,可以使用np.squeeze()函数进行压缩。
希望对你有所帮助!如果还有其他问题,请随时提问。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![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)