Python——用os模块寻找指定目录(包括子目录)下所有图片文件

import os  # 导入os模块

def search_file(start_dir):
    img_list = []
    extend_name = ['.jpg', '.png', '.gif']  # 图片格式，可以添加其他图片格式
    os.chdir(start_dir)  # 改变当前工作目录到指定的路径

    for each_file in os.listdir(os.curdir):
        # listdir()返回指定的文件夹包含的文件或文件夹的名字的列表 curdir表示当前工作目录

        img_prop = os.path.splitext(each_file)
        if img_prop[1] in extend_name:
            img_list.append(os.getcwd() + os.sep + each_file + os.linesep)
            # os.getcwd()获得当前路径 os.sep分隔符 os.linesep换行符

        if os.path.isdir(each_file):  # isdir()判断是否为文件夹
            search_file(each_file)  # 递归搜索子文件夹下的图片
            os.chdir(os.pardir)  # 返回上一级工目录

    with open(r'C:\Users\crazy\Desktop\img_list.txt', 'a') as file_obj:  # 此处修改输出文本文件目录及名称
        file_obj.writelines(img_list)  # writelines按行写入数据

if __name__ == '__main__':
    start_dir_list = [r'C:\Users\crazy\Documents\HBuilderProjects\瀑布流\img',
                      r'D:\Pic']
    for each_dir in start_dir_list:
        search_file(each_dir)

　　附上结果图：