Python读取文件夹所有指定类型文件

代码

利用os.walk()找到所有文件名，再利用string.endswith()判断文件类型，就能提取出来了

import os
# directory-主路径
# fileType-指定文件类型
# fileList-目标类型文件列表（路径+文件名）
def SearchFiles(directory, fileType):      
    fileList=[]    
    for root, subDirs, files in os.walk(directory):
        for fileName in files:
            if fileName.endswith(fileType):
                fileList.append(os.path.join(root,fileName))
    return fileList

使用

directory = 'mpii/'
fileType '.jpg'
fileList = SearchFiles(directory, fileType)
print(fileList)