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)))(数组([0, 1, 1, 1, 2, 2, 2], dtype=int64), 数组([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])