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

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'文件夹来实现的。我们使用Python自带的imghdr模块来检查文件是否是图片。你还可以通过查看imghdr.what()函数返回的结果，来过滤掉你不感兴趣的图片类型。