如何遍历目录子文件夹并删除其中非.py后缀的文件

你推测的没错，os.listdir() 仅返回目标目录下的条目名称字符串，需要将名称和父目录拼接为完整路径后，才能判断条目的实际类型。以下提供两种常用实现方案：

方案1：使用os标准库实现 #

import os

# 替换为你实际的根目录路径
root_dir = "./dir"

# 遍历根目录下的所有条目
for entry_name in os.listdir(root_dir):
    # 拼接得到条目完整路径
    entry_full_path = os.path.join(root_dir, entry_name)
    # 判断是否为文件夹
    if os.path.isdir(entry_full_path):
        # 遍历该子文件夹下的所有文件
        for file_name in os.listdir(entry_full_path):
            file_full_path = os.path.join(entry_full_path, file_name)
            # 仅处理文件、跳过子文件夹（如果需要处理嵌套文件夹可自行调整逻辑）
            if os.path.isfile(file_full_path):
                # 判断后缀是否不是.py
                if not file_name.endswith(".py"):
                    # 执行删除，操作前建议先print(file_full_path)确认要删除的文件列表
                    os.remove(file_full_path)

方案2：使用pathlib实现（Python3.4及以上版本支持，语法更简洁） #

from pathlib import Path

# 替换为你实际的根目录路径
root_dir = Path("./dir")

# 遍历根目录下的所有文件夹
for sub_folder in root_dir.iterdir():
    if sub_folder.is_dir():
        # 遍历子文件夹下的所有文件
        for file in sub_folder.iterdir():
            if file.is_file() and file.suffix != ".py":
                # 执行删除，操作前建议先print(file)确认要删除的文件列表
                file.unlink()

注意：正式执行删除操作前，建议先将删除代码替换为打印待删除文件路径的逻辑，确认匹配规则无误后再执行删除，避免误删重要文件。

如果需要处理子文件夹下的嵌套子目录，可以将遍历逻辑替换为递归遍历，或使用os.walk()接口实现。

内容的提问来源于stack exchange，提问作者AMM