我有一個存儲在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的“花式索引”,這讓我非常困惑……我嘗試解決這個問題的大多數嘗試都是過度的,而且不必要的是巴洛克式的.

我確信有一個非常簡單的方法,我在這里忽略.這樣做的最佳方法是什么 – 獲得這些價值的機制是否比我概述的更好？

總結

以上是生活随笔為你收集整理的python ndarray find_python – 从存储在NumPy ndarrays中的图像中查找特定(R,G,B)颜色值的(x,y)索引...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。