是否可以用python检测像素颜色？

Question

我试图通过颜色来检测一个确切的像素，然后无论是否找到该像素都做一些事情。我认为你（显然）不能用 vanilla python 来做这个，但我还没有找到任何模块来做到这一点。如果这是可能的，我想举一些例子。感谢您的建议。

编辑：基本上，我需要扫描一个像素（1920x1080 屏幕）并获取 RGB 值

Answer 1

您可以使用Pillow 做您想做的事。

您可以尝试以下功能。这是https://stackoverflow.com/a/765774/8581025代码的修改版本。

from PIL import Image

def detect_color(rgb, filename):
    img = Image.open(filename)
    img = img.convert('RGBA')
    data = img.getdata()

    for item in data:
        if item[0] == rgb[0] and item[1] == rgb[1] and item[2] == rgb[2]:
            return True
    return False

比如想知道当前工作目录下example.png文件中是否有红色像素（255,0,0），可以使用如下代码。

detect_color((255, 0, 0), 'example.png')  # returns True if there is a red pixel, False otherwise.

获取特定坐标的像素颜色，可以参考Get pixel's RGB using PIL。

Answer 2

您可以使用numpy 和pyautogui：

import numpy as np
import pyautogui

image = np.array(pyautogui.screenshot())

np.argwhere(image == [255, 255, 255])

它将返回所有具有指定颜色的点。