Numpy.argwhere
时间: 2023-11-18 11:57:27 浏览: 54
numpy.array 操作使用简单总结
Numpy.argwhere 函数返回一个数组中非零元素的索引,这些索引可以是多维的。例如,如果有一个二维的数组 arr,那么 numpy.argwhere(arr) 将返回一个包含非零元素索引的二维数组,其中每个非零元素索引均表示为一个长度为 2 的元组。下面是一个示例:
```python
import numpy as np
arr = np.array([[0, 1, 0], [1, 0, 0], [0, 0, 1]])
print(arr)
# Output:
# [[0 1 0]
# [1 0 0]
# [0 0 1]]
indices = np.argwhere(arr)
print(indices)
# Output:
# [[0 1]
# [1 0]
# [2 2]]
```
在上面的示例中,我们首先创建了一个二维的 numpy 数组 arr,并打印了它的值。然后,我们使用 numpy.argwhere 函数获取了 arr 中的非零元素索引,并将它们存储在 indices 变量中。最后,我们打印了 indices 的值,它包含了 arr 中所有非零元素的索引。
阅读全文