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])