如何获取第一维和第二维的值，其中第三维是numpy数组中的特定值

Question

在2D NumPy数组中，如何获取第三个元素为255的那些行的第一个和第二个元素？

arr = np.array([[ 32, 32, 255], [21, 21, 10], [25, 27, 18]], dtype=np.uint8)

如何找到第一个和第二个元素，在这种情况下都是32？

我基本上想要获得包含特定颜色像素的坐标。

Answer 1

一些切片工作：

>>> arr[arr[:,-1] == 255][:, :2]
array([[32, 32]], dtype=uint8)