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

Question

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

我可以通过使用for语句手动运行行和列来完成此操作，但这看起来相当慢，而且我正面有更好的方法来执行此操作。

我试图修改我在本文中找到的解决方案。 Finding the (x,y) indexes of specific (R,G,B) color values from images stored in NumPy ndarrays

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 documentation

Answer 1

我不太明白的问题。 all()中的轴参数在颜色通道（轴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])) 
>>> 

这其实只是重复您链接到答案。但是评论太长了。也许你可以解释一下为什么你需要修改以前的答案？看起来你并不需要。