python - 如何使用 pygame 从 python 中用户指定的文件夹导入所有图像

Question

我是 python 新手，正在研究 pygame。对于我的项目，我需要从用户指定的文件夹导入所有图像。 pygame 中是否有任何函数可以仅从文件夹中导入图像文件？或指导我仅过滤导入文件夹中所有文件中的图像文件。抱歉，如果问题太基本了。

Answer 1

我不了解 pygame，但使用普通的 python 就可以相当简单地获取文件夹中的所有图像文件:

import imghdr
import os

for dirpath, dirnames, filenames in os.walk('FOLDER'):
    for filename in filenames:
        file_path = os.path.join(dirpath, filename)
        if imghdr.what(file_path):
            print file_path, ' is an image'

这是通过递归探索“FOLDER”来实现的。我们使用 imghdr 内置模块检查文件是否是图像。您甚至可以通过检查 imghdr.what() 函数返回的内容来过滤掉您不感兴趣的图像类型。