在python数组中查找值的X和Y坐标

Question

我正在尝试在numpy图像数组中获取给定值的x和y坐标。

我可以通过使用for语句手动遍历行和列来做到这一点，但这似乎相当慢，并且我认为有更好的方法可以做到这一点。

我正在尝试修改本文中找到的解决方案。 从存储在NumPy ndarrays中的图像中查找特定（R，G，B）颜色值的（x，y）索引

a = image
c = intensity_value

y_locs = np.where(np.all(a == c, axis=0))
x_locs = np.where(np.all(a == c, axis=1))

return np.int64(x_locs), np.int64(y_locs)

我有np.int64将值转换回int64。

我也在看numpy.where文档

Answer 1

我不太明白这个问题。 all()的axis参数在颜色通道（第2轴或-1轴）上而不是在x和y索引上运行。 然后where()将为您提供图像中匹配值的坐标：

>>> # set up data
>>> image = np.zeros((5, 4, 3), dtype=np.int)
>>> image[2, 1, :] = [7, 6, 5]
>>> # find indices
>>> np.where(np.all(image == [7, 6, 5], axis=-1))
(array([2]), array([1]))
>>>

这实际上只是重复您链接的答案。 但是评论太久了。 也许您可以再解释一下为什么您需要修改先前的答案？ 似乎不需要。