Skip to main content
 首页 » 编程设计

python之有没有办法检查列表是否在 np.array 中

2025年05月04日17dudu

我想找到一种方法来检查我的图像是否具有任何特定的 RGB 值。

我有一个 list

rgb = [128,134,45]

和一张图片

img = array([[[0,0,0],[128,134,45],[128,134,45],...,[0,0,0],[0,0,0 ],[0,0,0]]]) 形状 =(64,64,3)

并且想要获得 rgb 匹配 img[j][i] 的位置 (i,j) 例如 pos = [(0,1),(0,2),...]

我尝试使用“in”,如 if rgb in img: brbr 但没有用。

请您参考如下方法:

>>>np.where((img==rgb).all(axis=2)) 
(array([0, 0], dtype=int32), array([1, 2], dtype=int32)) 

这为您提供了 rgb 值的索引

img[0][1] 和 img[0][2]