Python使用os.walk遍历目录时如何仅获取指定目标根目录路径

核心问题说明 #

你当前编写的遍历代码存在两个导致输出不符合预期的问题：

• 接收os.walk返回值时没有拆包，默认拿到的是完整三元组，没有单独提取路径字段做匹配判断
• 没有控制os.walk的递归深度，默认会无差别遍历从根目录到最深层的所有嵌套目录，把.git、src等深层路径全部纳入输出范围

具体调整方法 #

os.walk每次迭代会返回固定结构的三元组：(当前遍历的绝对/相对路径, 当前路径下的所有子目录名称列表, 当前路径下的所有文件名称列表)，可以基于这个结构做两层控制：

• 路径匹配：按照你要抓取的CODE_BASE/gateway-developer-example这类目标的特征，判断当前遍历到的路径是否符合要求
• 递归控制：修改返回的子目录列表，让os.walk不需要深入无意义的深层路径遍历，减少无效计算

场景1：抓取根目录下第一层的项目类路径 #

如果你要的目标路径就是传入的根目录cbdir下直接嵌套的一层子目录（比如cbdir下的gateway-developer-example，不需要抓它内部的.git、global/src等内容），可以直接用下面的代码实现：

import os

cbdir = "CODE_BASE"
target_path_list = []

for root, dirs, files in os.walk(cbdir):
    # 计算当前路径相对于根目录的路径差
    current_rel_path = os.path.relpath(root, cbdir)
    # 相对路径中不存在路径分隔符，说明就是根目录下的直接子目录，符合目标要求
    if os.sep not in current_rel_path:
        target_path_list.append(root)
    # 关键控制逻辑：清空当前层的子目录列表，告诉os.walk不要继续向内层递归
    dirs[:] = []

场景2：按命名规则跨深度匹配目标路径 #

如果你不需要严格限制目录层级，只要目录名符合特征（比如目录名带-developer-example后缀）就抓取，可以调整匹配逻辑：

import os

cbdir = "CODE_BASE"
target_path_list = []

for root, dirs, files in os.walk(cbdir):
    current_dir_name = os.path.basename(root)
    # 自定义匹配规则：抓取所有名称以-developer-example结尾的目录
    if current_dir_name.endswith("-developer-example"):
        target_path_list.append(root)
        # 匹配到目标路径后，停止向内层递归，不需要遍历目录下的.git、编译产物等内容
        dirs[:] = []

注意事项 #

• 必须对os.walk的返回值做拆包处理，也就是写成for root, dirs, files in os.walk(...)，不能像之前那样直接用一个变量接收，否则无法拿到单独的路径、子目录字段做逻辑处理
• 修改dirs列表时必须用切片赋值的方式（比如dirs[:] = []），不能直接给dirs赋空值，否则os.walk感知不到你对子目录遍历范围的调整

os.walk的底层机制会在每次遍历到当前目录时，读取你修改后的dirs列表决定下一层要遍历哪些子目录，只有直接修改原列表内容，才能生效递归控制逻辑

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