欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python3获取指定格式文件列表

程序员文章站 2022-03-31 09:37:38
...
import os

def get_file_list(search_dir:str, dFormat:str) -> list:    
    """在指定路径搜索指定类型文件
    
    Arguments:
        search_dir {str} -- 搜索路径,注意Windows下要使用\\或者/
        dFormat {str} -- 要搜索的文件格式,例如.pdf
    
    Returns:
        list -- 返回数组对象
    """
    os.chdir(search_dir)
    files = filter(os.path.isfile, os.listdir(search_dir))
    files = [os.path.join(search_dir, f) for f in files]
    # 将文件列表按照创建时间倒序排列
    # files.sort(key=lambda x: os.path.getctime(x),reverse=True)
    
    t = []
    for f in files:
        if(os.path.isfile(f)):
            if(os.path.splitext(f)[1] == dFormat):
                t.append(f)        
    return(t)