如何从图像文件> numpy数组>单一RGB颜色的x / y坐标列表转换

Question

Create a numpy array from a 2x2 pixel image above (zoomed in for clarity):

import numpy as np
from PIL import Image

img = Image.open('2x2.png')
pixels = np.array(img)

Array looks like this, with each pixel represented by its respective [R, G, B] values:

>>> pixels
array([[[255,   0,   0],
        [  0, 255,   0]],

       [[  0,   0, 255],
        [255,   0,   0]]], dtype=uint8)

Now I need to produce an array of x/y coordinates of 'all the red pixels', so all array elements with value [255, 0, 0] . The resulting array of coordinates needed looks like this:

array([[ 0,  0],
       [ 1,  1 ]])

What's the best way to achieve this?

Answer 1

You can try:

temp = (pixels == [255,0,0]).all(axis=-1)
# [[ True False]
#  [False  True]]
result = np.asarray(np.where(temp)).T
print(result)

# print
# [[0 0]
#  [1 1]]

Answer 2

I found that this works:

np.argwhere((pixels==[255,0,0]).all(axis=2))
array([[0, 0],
       [1, 1]]