Python如何避免ImageCollection类两个图片加载函数的for循环重复？

1. 核心方案：提取公共逻辑，抽象差异条件 #

你的两个思路（辅助函数、回调）都是正确的，核心是把遍历文件夹、加载图片、存入字典这些重复代码抽离成公共逻辑，仅将判断文件是否需要加载的差异部分作为可配置参数传入。

示例实现：

import os
import pygame

class ImageCollection:
    def __init__(self):
        self.images = {}
    
    # 私有公共方法：封装所有重复的加载逻辑
    def _load_images(self, folder_path, should_load):
        for filename in os.listdir(folder_path):
            # 先过滤非图片文件
            if not filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
                continue
            # 用传入的判断函数决定是否加载当前文件
            if should_load(filename):
                img_path = os.path.join(folder_path, filename)
                try:
                    img = pygame.image.load(img_path)
                    # 用无后缀文件名作为字典键
                    img_name = os.path.splitext(filename)[0]
                    self.images[img_name] = img
                except pygame.error:
                    print(f"加载图片失败：{img_path}")
    
    # 场景1：加载文件夹内所有图片
    def load_all(self, folder_path):
        self._load_images(folder_path, lambda filename: True)
    
    # 场景2：加载包含指定字符串的图片
    def load_by_keyword(self, folder_path, keyword):
        # 用lambda捕获外部参数，解决回调带额外参数的问题
        self._load_images(folder_path, lambda filename: keyword in filename)

2. 你的思路验证 #

• 辅助函数：如果仅抽离「添加字典条目」的逻辑，只能减少局部重复，不如直接抽离整个加载流程彻底。但如果加载逻辑中还有其他重复小步骤（比如统一缩放图片、转换颜色格式），辅助函数依然是有效的补充。
• 回调带参数：之前受阻的参数问题，用lambda或functools.partial就能解决。比如用partial的写法：

from functools import partial

def _match_keyword(filename, keyword):
    return keyword in filename

def load_by_keyword(self, folder_path, keyword):
    self._load_images(folder_path, partial(_match_keyword, keyword=keyword))

3. 进阶扩展：支持更灵活的筛选 #

这种抽象方式可以轻松扩展到更复杂的筛选场景，比如正则匹配、按文件大小筛选：

# 示例：按正则表达式匹配文件名
def load_by_regex(self, folder_path, pattern):
    import re
    regex = re.compile(pattern)
    self._load_images(folder_path, lambda filename: regex.search(filename) is not None)

4. 额外建议 #

• 在公共加载方法中加入错误捕获，避免单个损坏图片导致整个加载流程中断；
• 如果需要重复加载同一文件夹，可以加入缓存逻辑，记录已加载的文件夹路径，避免重复IO操作。

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