python - 有没有办法在二维数组中找到输入值的每个索引？

Question

我一直在尝试在 python 中创建一个函数，该函数返回二维数组中每个重复出现的值的 x、y 坐标。例如，如果我有数组和一个值，

array = [ [1 ,2 ,3]
          [2 ,3 ,1]
          [3 ,2, 1]]

search = 1

它会输出(0,0) (1,2) (2,2)

我一直在尝试使用一些函数，例如 np.where 或将其转换为 pandas 数据框并以这种方式进行搜索，但我不确定最好的方法。当我使用 np.where 时，它​​返回一个空数组，因为我使用的是长小数。我正在尝试在 200 x 200 的数组上执行此操作。

Answer 1

我们可以做 np.where PS:a 是你的数组

list(zip(*np.where(a==search)))
[(0, 0), (1, 2), (2, 2)]

正如 hpaulj 提到的那样

np.argwhere(np.isclose(a,search))
array([[0, 0],
       [1, 2],
       [2, 2]])