如何通过列表推导式结合PyAutoGUI识别指定图片并跳转至对应处理函数？

嘿，这个需求很清晰，我们可以通过两种直观的方式实现你想要的逻辑——要么基于你已经生成的number列表做索引映射，要么直接在检测过程中匹配图片名称与函数，两种方法都能精准定位到被识别的图片并调用对应函数。下面给你详细拆解：

方法一：基于现有number列表的索引映射 #

你已经用列表推导式生成了检测结果列表number，其中只有一个元素不为None。我们只需要找到这个非None元素的索引，再通过索引匹配对应的函数即可：

import pyautogui

def one(): 
    print("执行one函数：检测到one.png")
def two(): 
    print("执行two函数：检测到two.png")
def three(): 
    print("执行three函数：检测到three.png")
def four(): 
    print("执行four函数：检测到four.png")

# 生成检测结果列表
number = [pyautogui.locateOnScreen(f'{nr}.png', confidence=0.95) for nr in ('one', 'two', 'three', 'four')]

# 找到第一个非None的元素索引（因为你说仅能识别到一张，所以第一个就是唯一的）
detected_index = next((idx for idx, result in enumerate(number) if result is not None), None)

if detected_index is not None:
    # 定义函数映射列表，顺序必须和图片检测的顺序完全一致
    func_list = [one, two, three, four]
    func_list[detected_index]()  # 调用对应函数
else:
    print("未检测到任何目标图片")

关键逻辑说明： #

• enumerate(number)会同时返回每个检测结果的索引和值，方便我们定位位置；
• next(...)用来获取第一个符合条件（非None）的索引，因为你明确只有一张图片会被识别，所以不需要遍历全部；
• func_list的顺序必须和列表推导式里的('one', 'two', 'three', 'four')顺序一致，确保索引和函数一一对应。

方法二：直接通过图片名称映射函数（更简洁） #

我们可以把检测和函数匹配合并成一步，不用先存number列表，直接通过图片名称作为键来映射对应的函数，可读性更强：

import pyautogui

def one(): 
    print("执行one函数：检测到one.png")
def two(): 
    print("执行two函数：检测到two.png")
def three(): 
    print("执行three函数：检测到three.png")
def four(): 
    print("执行four函数：检测到four.png")

# 建立图片名称与函数的映射关系
image_func_map = {
    'one': one,
    'two': two,
    'three': three,
    'four': four
}

# 遍历图片名称，找到第一个检测成功的
detected_image = next((name for name in image_func_map if pyautogui.locateOnScreen(f'{name}.png', confidence=0.95) is not None), None)

if detected_image:
    image_func_map[detected_image]()  # 调用对应函数
else:
    print("未检测到任何目标图片")

优势： #

• 不用维护索引和函数的顺序关系，直接通过名称绑定，逻辑更直观；
• 减少了中间变量number的存储，代码更简洁。

两种方法都能完美实现你的需求，你可以根据自己的代码习惯选择~

内容的提问来源于stack exchange，提问作者Bramsko