如何通过Python3的os.walk获取并打印当前目录与迭代中的下一级目录？

嘿，我来帮你理清楚怎么用os.walk处理当前目录和下一级目录的需求～

首先得明确os.walk的工作机制：它默认会递归遍历所有子目录，从指定的根目录开始，每次迭代返回一个三元组(root, dirs, files)，其中：

• root是当前正在遍历的目录路径
• dirs是当前root目录下的所有直接子目录（也就是你要的下一级目录）
• files是当前root目录下的所有文件

所以你要获取当前目录和它的下一级目录，其实不用额外调用next(os.walk(root))，因为dirs已经直接给出了当前目录的下一级目录列表。

1. 遍历指定目录，打印当前目录及下一级目录 #

如果你只是想遍历指定目录（比如/tmp/test），并打印每个当前目录和它的直接子目录，直接用dirs字段就好，不需要嵌套调用os.walk。这里给你修正后的代码：

import os

currentdir = '/tmp/test'
for root, dirs, files in os.walk(currentdir):
    print(f"Current dir: {root}")
    # dirs就是当前root的下一级目录列表，如果没有子目录就是空列表
    print(f"Next level dirs: {dirs}")
    
    # 如果你只想遍历当前目录和它的直接子目录（不递归更深），可以清空dirs来终止递归
    # 去掉下面这行注释的话，就只会处理currentdir和它的第一级子目录，不会继续往下走
    # dirs[:] = []

解释一下：

• 如果你需要只处理当前目录和它的下一级目录（不递归更深），就加上dirs[:] = []这行，因为os.walk会根据dirs的内容决定要不要继续遍历子目录，清空它就会终止递归。
• 如果你需要递归遍历所有目录，但每次都打印当前目录和它的直接下一级目录，就去掉那行注释，这样会遍历所有层级，每个层级都输出当前目录和它的子目录。

2. 你的原代码问题说明 #

你原来的代码里next(os.walk(root))会有两个问题：

1. next(os.walk(root))返回的是(root, dirs, files)三元组，直接打印的话格式会很混乱，不是你想要的“下一级目录”。
2. 如果当前root目录没有子目录，os.walk(root)会只返回一次迭代，next()之后再调用会抛出StopIteration异常。

所以完全没必要嵌套调用os.walk，直接用每次迭代返回的dirs就足够了。

3. 只获取当前目录和它的直接子目录（不递归）的另一种写法 #

如果你不需要遍历文件，只是想获取当前目录和它的下一级目录路径，也可以用os.listdir结合os.path.isdir来实现，但os.walk的方式更简洁：

import os

currentdir = '/tmp/test'
# 先打印当前目录
print(f"Current dir: {currentdir}")
# 获取当前目录的下一级目录
next_level_dirs = [
    os.path.join(currentdir, d) 
    for d in os.listdir(currentdir) 
    if os.path.isdir(os.path.join(currentdir, d))
]
print(f"Next level dirs: {next_level_dirs}")

这种方式更直接，但如果需要同时处理文件的话，还是os.walk更方便。

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