天天看点

python获取过滤文件夹下所有指定格式的文件

获取文件夹内所有文件名,过滤指定格式扩展名的文件

def get_files() -> list:
    file_list = os.listdir('.')  ## 获取当前文件夹内所有文件名
    
    for file in file_list[::-1]:  ## 逆序遍历
        if file.endswith('.xlsx'):  ## 判断文件的扩展名
            continue
        file_list.remove(file)  ## 过滤文件
    
    return file_list  ## 返回文件名列表