python 文件夹结构可视化

Python 有一些库可以帮助我们可视化文件夹结构。其中一个常用的库是 `os` 模块，它提供了一些用于处理文件和目录的函数。另一个常用的库是 `graphviz`，它可以生成各种图形，包括文件夹结构。

在使用 `os` 模块时，我们可以使用函数 `os.walk` 来遍历文件夹中的所有文件和子文件夹。这个函数返回一个生成器，我们可以使用循环来迭代它，并获取文件夹中的所有项。然后，我们可以使用字符串操作和缩进来打印文件夹结构。以下是一个简单的示例：

import os

def print_directory_structure(folder_path, indent=''):
    for root, dirs, files in os.walk(folder_path):
        print(indent + os.path.basename(root) + '/')
        for filename in files:
            print(indent + '├──' + filename)
        for dirname in dirs:
            print_directory_structure(dirname, indent + '│   ')

# 调用函数并传入文件夹路径
print_directory_structure('path/to/folder')

上面的代码会以树形结构打印文件夹中的所有文件和子文件夹。其中文件夹会使用斜杠“/”表示，文件会使用框线与文件名表示。

如果想要更加美观和可交互的文件夹结构可视化，可以使用 `graphviz` 模块。这个模块可以生成各种图形，包括树形结构。下面是一个简单的示例：

import os
from graphviz import Digraph

def draw_directory_structure(folder_path, graph):
    for root, dirs, files in os.walk(folder_path):
        graph.node(root)
        for filename in files:
            graph.node(filename)
            graph.edge(root, filename)
        for dirname in dirs:
            draw_directory_structure(dirname, graph)
            graph.edge(root, dirname)

# 创建一个空的图形对象
graph = Digraph()

# 调用函数并传入文件夹路径和图形对象
draw_directory_structure('path/to/folder', graph)

# 展示图形
graph.view()

上面的代码使用 `graphviz` 创建了一个空的图形对象，并通过递归方式遍历文件夹结构，将文件夹和文件作为节点，通过边连接起来。最后，调用 `view` 方法可以在新的窗口中打开图形。这样就可以以更加美观的方式显示文件夹结构。