我有一个存储在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的“花式索引”,这让我非常困惑……我尝试解决这个问题的大多数尝试都是过度的,而且不必要的是巴洛克式的.
我确信有一个非常简单的方法,我在这里忽略.这样做的最佳方法是什么 – 获得这些价值的机制是否比我概述的更好?