当前位置: 首页 > 后台 > Python

Python递归遍历图片文件

作者: admin 2019-07-12 12:01:51 阅读(337) 评论(0)
img_list = []  # 图片列表
# 递归寻找指定目录下所有jpg和png文件,并返回所有图片路径
def searchDirFile(rootDir):  # rootDir为遍历的文件夹名称
    for dir_or_file in os.listdir(rootDir):
        filePath = os.path.join(rootDir, dir_or_file)
        # 判断是否为文件
        if os.path.isfile(filePath):
            # 如果是文件再判断是否以.jpg结尾,不是则跳过本次循环
            if os.path.basename(filePath).endswith(('.jpg', '.png')):
                # 将图片路径写入图片列表
                img_list.append(filePath)
            else:
                continue
        # 如果是个dir,则再次调用此函数,传入当前目录,递归处理。
        elif os.path.isdir(filePath):
            searchDirFile(filePath)
        else:
            print('不存在此文件' + os.path.basename(filePath))
    return img_list  # 返回图片列表
0 条评论