如何自定义 os.walk() 的输出以可视化 Python 中的递归目录结构？

Python 中使用 os.walk() 进行递归目录遍历

在 Python 中，os.walk() 函数是一个强大的工具递归遍历目录和子目录。通过利用其功能，您可以在文件系统中导航并高效地访问文件和目录。

使用 os.walk() 打印目录结构

考虑以下内容code:

import os

for root, dirs, files in os.walk("."):
    print(root)
    print("")
    for items in fnmatch.filter(files, "*"):
        print("..." + items)
    print("")

此代码将遍历当前目录并打印其中的目录和文件。但是，输出不会采用所需的格式：目录和文件将以相同的方式列出。

自定义目录遍历输出

打印目录和根据需要的文件结构，如：

A
---a.txt
---b.txt
---B
------c.out

其中A、B为目录，其余为文件，可以修改代码如下：

import os

for root, dirs, files in os.walk("."):
    path = root.split(os.sep)
    print((len(path) - 1) * '---', os.path.basename(root))
    for file in files:
        print(len(path) * '---', file)

修改后的代码：

• 使用 os.sep 分隔符将路径拆分为列表，该分隔符表示当前操作系统上的目录分隔符。
• 根据路径组件的数量减 1。
• 使用缩进和目录或文件的基本名称的组合，以所需的格式显示结构。

通过调整缩进并使用basename 函数，您可以自定义输出以满足您的特定要求。这种技术可以让您可视化目录层次结构并更方便地访问文件和目录。

以上是如何自定义 os.walk() 的输出以可视化 Python 中的递归目录结构？的详细内容。更多信息请关注PHP中文网其他相关文章！