os.walk循环未将层级最后一个目录的文件存入列表的问题

问题场景与代码 #

在Windows 10系统中，使用os.walk和os.path.splitext生成指定磁盘的文件夹名列表和无扩展名文件名列表，代码如下：

import os

output_files = []
output_dir = []

ex_dir = set(['dir1','dir2'])
inc_ext = set(['ext1','ext2'])


for root, names_dir, names_file in os.walk(root):
    names_dir[:] = [d for d in names_dir if d not in ex_dir]
    names_file = [file for file in names_file if os.path.splitext(file)[1] in inc_ext]
    output_dir.extend(names_dir)
    output_files.extend(names_file)

遇到的问题：文件夹名称收集正常，但最后一个目录的文件名始终无法存入output_files，已排除权限问题，添加无效的continue语句也未解决。

排查与解决思路 #

1. 扩展名匹配逻辑错误 #

os.path.splitext(file)[1]返回的扩展名带英文点号（例如.ext1），但你的inc_ext集合里是不带点的ext1，这会导致匹配条件不成立。若前面目录的文件被收集可能是代码书写失误，最后一层刚好触发了这个问题。

• 修正方案：
  将inc_ext改为带点的扩展名集合：

  inc_ext = set(['.ext1', '.ext2'])
  

  若需兼容Windows文件名的大小写差异，可统一转小写判断：

  names_file = [file for file in names_file if os.path.splitext(file)[1].lower() in (ext.lower() for ext in inc_ext)]
  

2. 变量名冲突导致逻辑混淆 #

你将os.walk的起始路径参数和循环变量都命名为root，虽然Python函数参数在调用时就已求值，但这种命名极易引发逻辑误解，建议修改起始路径变量名：

start_path = "C:\\"  # 替换为你的目标磁盘路径
for root, names_dir, names_file in os.walk(start_path):

3. 无效的continue语句需删除 #

你添加的if names_file is not None: continue完全无效——names_file永远是列表（空列表也不等于None），反而可能跳过后续的extend操作，直接删除该语句即可。

4. 添加调试日志定位问题 #

若以上方案未解决，可在循环内添加调试打印，明确最后一层目录的文件处理情况：

for root, names_dir, names_file in os.walk(start_path):
    names_dir[:] = [d for d in names_dir if d not in ex_dir]
    # 调试打印：查看当前目录和文件处理情况
    print(f"当前遍历目录：{root}")
    print(f"原始文件列表：{names_file}")
    names_file = [file for file in names_file if os.path.splitext(file)[1] in inc_ext]
    print(f"过滤后待收集文件：{names_file}")
    output_dir.extend(names_dir)
    output_files.extend(names_file)

补充：实现无扩展名文件名收集 #

当前代码收集的是带扩展名的文件名，若需无扩展名的文件名，修改extend逻辑：

output_files.extend([os.path.splitext(file)[0] for file in names_file])

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