numpy返回值为0元素的索引
时间: 2024-03-06 10:50:44 浏览: 59
numpy中实现ndarray数组返回符合特定条件的索引方法
在 NumPy 中,返回值为 0 的元素的索引可以通过 `numpy.where()` 函数来获取。例如,对于一个形状为 (3, 4) 的数组 `arr`,可以通过以下代码获取到值为 0 的元素的索引:
```
import numpy as np
arr = np.array([[1, 2, 3, 4],
[5, 6, 0, 8],
[9, 0, 11, 12]])
zero_indexes = np.where(arr == 0)
print(zero_indexes)
```
运行结果为:
```
(array([1, 2]), array([2, 1]))
```
其中,结果是一个元组,包含两个一维数组。第一个数组表示行索引,第二个数组表示列索引。在上面的例子中,值为 0 的元素分别位于第 2 行第 3 列和第 3 行第 2 列。
阅读全文