os.walk 但顶部有目录？

Question

我有一些简单的代码可以打印出目录的结构。 我的示例目录ABC 包含子目录A，其中包含A.txt，一个子目录Z 包含Z.txt，以及一个文件info.txt。在实际使用中，这将是许多文件和嵌套目录的大集合。

import os
topdir = 'ABC/'
for dirpath, dirnames, files in os.walk(topdir):
    print(os.path.join(dirpath))
    for name in files:
        print(os.path.join(dirpath, name))

输出是：

ABC/
ABC/info.txt
ABC/A
ABC/A/A.txt
ABC/Z
ABC/Z/Z.txt

如何才能使目录在顶部处理/打印？ 我希望输出复制我在 Windows 资源管理器中看到的内容，它首先显示目录，然后显示文件。

我想要的输出：

ABC/
ABC/A
ABC/A/A.txt
ABC/Z
ABC/Z/Z.txt
ABC/info.txt

Answer 1

无需将所有文件存储在列表中并以一种或另一种方式对该列表进行排序，您可以创建一个递归函数并首先递归到目录结构的下一级，然后再打印当前级别的文件：

def print_dirs(directories):
    try:
        dirpath, dirnames, files = next(directories)
        print(dirpath)              # print current path; no need for join here
        for _ in dirnames:          # once for each sub-directory...
            print_dirs(directories) # ... recursively call generator
        for name in files:          # now, print files in current directory
            print(os.path.join(dirpath, name))
    except StopIteration:
        pass

print_dirs(os.walk(topdir))

堆栈也可以这样做，但我认为这样会更清晰一些。是的，这也会将一些目录存储在列表中/堆栈上，但不是所有文件，而是与嵌套目录的级别一样多。

编辑：这有在生成器上打印 any 下一个目录的问题，即使那不是子目录而是兄弟（或“叔叔”或其他）。 for _ in dirnames 循环应该解决这个问题，如果有的话，对每个子目录进行一次递归调用。目录本身不必作为参数传递，因为它会从生成器中获取。