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）的索引：

  

    

array（[1,2]）

  

Answer 1

您正在与np.all一起寻找numpy.nonzero（以确保每个RGB都匹配）：

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