Numpy.where使用

Question

使用numpy.where獲取numpy.array中的所有（R，G，B），其中具有一定的R，G和B值

問題是我不確定我可以使用numpy.where獲得我想要的東西：

我嘗試了以下代碼：

L = numpy.array([[1,2,3],[1,1,1],[1,1,1]])
print(numpy.where(L==(1,1,1)))

（array（[0,1,1,1,2,2,2]，dtype = int64），array（[0,0,1,2,0,1,2]，dtype = int64））

我明白它返回我每個元素的坐標== 1但我希望它返回元素L中的索引等於（1,1,1）：

陣列（[1,2]）

Answer 1

您正在尋找numpy.nonzero和np.all （以確保每個RGB匹配）：

>>> numpy.nonzero(numpy.all(L == (1, 1, 1), axis=1))[0]
array([1, 2])