python - 列出二维数组 numpy 中具有特定值的所有 x,y 坐标

Question

我想找到一个比loop&if“更numpy”的解决方案，用于列出二维数组python中所有值等于给定m的坐标(x，y)的任务。

例如:这是一个 4x4 矩阵

0 1 1 0
0 2 2 0
0 2 1 0
0 0 0 0

如果 m = 2 ，我想要 [(1,1), (1,2), (2,1)] 列表，因为这些单元格 = 2。我想要的是它们的坐标。

如果 m = 1 则 [(0,1), (0,2), (2,2)] 由于单元格 = 1。

我不需要循环和 if 的解决方案并将 i,j 放入列表中。它有点慢，有任何使用 numpy 更快的解决方案吗？谢谢

有人建议我看一下这个 numpy get index where value is true，但我尝试了，但它不正确。

详细说明:

案例中的 np.where(np.any(e==1, axis=0) 给出:[1,2] 是的!同意

np.where(np.any(e==1, axis=1) 给出:[0,2] 是的!还可以但是它不会导致这样: [(0,1), (0,2), (2,2)] 因为行或列信息不够，

所以请不要低估这个问题并一次又一次地删除我的问题。我已经厌倦了

Answer 1

丢失np.any部分。像这样:

np.array(np.where(e==1)).T

外部np.array和转置.T只是为了以易于阅读的方式排列索引。