Python 目录下文件读取

❤️ 前段时间发现了一个 人工智能学习网站，通俗易懂，风趣幽默，分享一下给大家。👈🏻 学习链接

目录下文件读取

1. 不包含子目录下文件

import os

def load_file_dir():
    file_path_dir = "download_list_json"
    if not os.path.exists(file_path_dir):
        return
    file_list = []
    for file_name in os.listdir(file_path_dir):
        # 判断是否是目录
        if os.path.isdir(file_name):
            continue
        # 文件过滤
        if ".temp" not in file_name:
            continue
        if "download_list" not in file_name:
            continue
        file_list.append(f"{file_path_dir}/{file_name}")

2. 包含子目录下文件

import os

def load_file_dir():
    file_path_dir = "data"
    if not os.path.exists(file_path_dir):
        return
    file_list = []
    for root, folder_names, file_names in os.walk(file_path_dir):
        for file_name in file_names:
            if ".json" not in file_name:
                continue
            new_root = root.replace("\\", "/")
            file_path = f"{new_root}/{file_name}"
            file_list.append(file_path)