如何无需扫描整个文件夹即可获取目录中首个jpg/png格式的图片

你现有代码的问题是直接取了目录返回的第一个文件，没有做格式校验。你可以对返回的文件列表逐个做后缀校验，找到第一个符合jpg/png格式的就直接终止遍历，不需要扫描整个文件夹的剩余文件。

实现代码 #

import os

# 配置允许的图片后缀，自动兼容大小写格式
ALLOWED_IMAGE_EXTS = {"jpg", "png"}

def find_first_image(target_dir):
    for root, _, files in os.walk(target_dir):
        # 遍历当前目录下的文件，找到第一个符合要求的就直接返回，不继续遍历后续内容
        for filename in files:
            file_ext = filename.split(".")[-1].lower()
            if file_ext in ALLOWED_IMAGE_EXTS:
                return os.path.join(root, filename)
        # 不需要遍历子目录的话保留这行break，需要递归扫子目录就删除
        break
    # 目录下无符合要求的图片时返回None
    return None

# 调用示例
current_path = os.getcwd()
# 取当前目录下的第一个子文件夹
first_subfolder = next(os.walk(current_path))[1][0]
target_folder = os.path.join(current_path, first_subfolder)
first_image_path = find_first_image(target_folder)
print(first_image_path)

优化说明 #

• 无需扫描全文件夹：内层文件遍历只要匹配到符合要求的图片就直接return，不会遍历完所有文件
• 兼容性优化：用os.path.join拼接路径，适配Windows、macOS、Linux不同系统的路径格式；后缀统一转小写判断，兼容JPG、PNG这类大写格式
• 可灵活调整：如果需要递归扫描子目录下的第一张图片，删除代码中的break即可

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