有没有办法检查一下名单上有没有人np.array？

Question

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

我有一张单子

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，但没用。你知道吗

Answer 1

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

这将为您提供rgb值的索引

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