用python获取指定目录下的指定类型的所有文件

快速从本地文件夹中挑选某一类型的文件，可以通过文件的扩展名来实现。
今天给大家分享一个思路，通过文件的扩展名来挑选指定目录下的某种类型的所有文件的python实现方法。

import os


def get_excel_file(dir_path):
    file_list = []
    for root, dirs, files in os.walk(dir_path):
        for file in files:
            if file.endswith("xlsx") or file.endswith("csv"):
                file_list.append(os.path.join(root, file))

    return file_list


if __name__ == '__main__':
    path = '您的目标文件夹路径'    # 如："D:\\ABC\\BB"
    print(get_excel_file(path))