python ndarray find_python – 从存储在NumPy ndarrays中的图像中查找特定(R,G,B)颜色值的(x,y)索引...

我有一个存储在numpy数组中的图像,由imread()产生：

>>> ndim

array([[[ 0, 0, 0],

[ 4, 0, 0],

[ 8, 0, 0],

...,

[247, 0, 28],

[251, 0, 28],

[255, 0, 28]],

[[ 0, 255, 227],

[ 4, 255, 227],

[ 8, 255, 227],

...,

[247, 255, 255],

[251, 255, 255],

[255, 255, 255]]], dtype=uint8)

>>> ndim.shape

(512, 512, 3)

我想有效地找到具有特定颜色值的像素的(x,y)坐标(或坐标),例如,

>>> c

array([ 32, 32, 109], dtype=uint8)

>>> ndim[200,200]

array([ 32, 32, 109], dtype=uint8)

>>> ndim.T[0, 200, 200]

32

>>> ndim.T[1, 200, 200]

32

>>> ndim.T[2, 200, 200]

109

…在这种情况下,我知道(200,200)处的像素具有RGB值(32,32,109) – 我可以测试它.

我想要做的是查询ndarray的像素值并获取坐标.在上述情况下,推定函数find_pixel(c)将返回(200,200).

理想情况下,这个find_pixel()函数将返回一个坐标元组列表,而不仅仅是它找到的第一个值.

我看过numpy的“花式索引”,这让我非常困惑……我尝试解决这个问题的大多数尝试都是过度的,而且不必要的是巴洛克式的.

我确信有一个非常简单的方法,我在这里忽略.这样做的最佳方法是什么 – 获得这些价值的机制是否比我概述的更好？