Python 遍历指定类型的文件

方式一

使用 os.walk 遍历所有文件，使用 endswith 筛选出需要的文件。子目录中的文件也会被筛选出来。

import os


def traverse_files(dir_path, file_extension):
    for root, dirs, files in os.walk(dir_path, followlinks=True):
        for file in files:
            if file.endswith(file_extension):
                print(os.path.join(root, file))


traverse_files('E:/Resource', '.jpg')

方式二

使用 glob 筛出需要的文件。子目录中的文件不会被筛选出来。

from pathlib import Path


def traverse_files(dir_path, file_extension):
    for file in Path(dir_path).glob('*' + file_extension):
        print(file)


traverse_files('E:\Resource', '.jpg')