python获取目录及子目录下（指定后缀）的文件名或文件路径

最新推荐文章于 2024-04-11 11:12:47 发布

The_Lonely_Wind

最新推荐文章于 2024-04-11 11:12:47 发布

阅读量849

点赞数

分类专栏： python实用脚本文章标签： python

本文链接：https://blog.csdn.net/the_lonely_wind/article/details/129079618

版权

python实用脚本专栏收录该内容

6 篇文章 1 订阅

订阅专栏

该代码定义了一个函数getfiles，用于递归地在指定的data_path及其子目录下查找所有文件。如果提供了后缀名参数suffix，函数将只返回匹配该后缀的文件。函数可以返回文件名或带有完整路径的文件名，取决于with_path参数的设置。

摘要由CSDN通过智能技术生成

"""递归获取文件"""
import os


def getfiles(data_path, suffix=None, data_file_list=None, with_path=False):
    """
    获取目录及子目录下所有文件名或者文件路径
    :param data_path: 数据根目录
    :param suffix: 后缀名
    :param data_file_list: 返回的文件名或路径
    :param with_path: 是否返回路径
    :return: 文件名或文件路径
    """
    if data_file_list is None:
        data_file_list = []
    if suffix is not None:
        if not isinstance(suffix, list):
            suffix = [suffix]
    data_files = os.listdir(data_path)
    for data_file in data_files:
        data_file_path = os.path.join(data_path, data_file)
        if os.path.isdir(data_file_path):
            getfiles(data_file_path, suffix, data_file_list, with_path)
        else:
            if suffix is None:
                if with_path:
                    data_file_list.append(data_file_path)
                else:
                    data_file_list.append(data_file)
            else:
                _, data_suffix = os.path.splitext(data_file)
                data_suffix = data_suffix.split('.')[-1]
                if data_suffix in suffix:
                    if with_path:
                        data_file_list.append(data_file_path)
                    else:
                        data_file_list.append(data_file)
    return data_file_list