如何根据搜索返回numpy ndarray的索引？答案

【问题标题】：How to return the index of numpy ndarray based on search?如何根据搜索返回numpy ndarray的索引？
【发布时间】：2023-03-24 20:02:01
【问题描述】：

我有一个 numpy 二维数组，

import numpy as np
array1 = array([[ 1,  2,  1,  1],
   [ 2,  2,  2,  1],
   [ 1,  1,  1,  1],
   [1,  3, 1, 1],
   [1,  1, 1, 1]])

我想找到元素“3”并知道它的位置。所以，我可以试试

condition = array1 == 3

然后为了找到论据，我可以尝试

np.argwhere(condition)

如何找到 3 所在的索引？

【问题讨论】：

【解决方案1】：

你也可以使用where，返回一个坐标元组：

In [34]: np.where(array1==3)
Out[34]: (array([3]), array([1]))

【讨论】：