层次树状图 - Python

层次树状图是一种数据可视化的方式，可以方便地展示树形结构的层次关系。在 Python 中，我们可以使用第三方库 anytree 来实现生成层次树状图。

安装 anytree

在使用 anytree 之前，需要先安装该库。使用 pip 安装即可：

!pip install anytree

创建一个树形结构

要生成层次树状图，首先需要创建一个树形结构。我们可以使用 anytree 库提供的节点类 Node 来创建：

from anytree import Node

root = Node("A")  # 创建根节点 A
b_node = Node("B", parent=root)  # 创建子节点 B，其父节点为 A
c_node = Node("C", parent=root)  # 创建子节点 C，其父节点为 A
d_node = Node("D", parent=b_node)  # 创建子节点 D，其父节点为 B

这里创建了一棵包含四个节点的树形结构，其中 A 为根节点，B 和 C 为其子节点，D 是 B 的子节点。

生成层次树状图

有了树形结构后，我们就可以使用 anytree 库提供的 RenderTree 方法生成层次树状图了：

from anytree import RenderTree

for pre, _, node in RenderTree(root):
    print(f"{pre}{node.name}")

运行上述代码后，就会生成如下的层次树状图：

A
├── B
│   └── D
└── C

以上是使用 anytree 生成树形结构的基本方法，更多高级用法和设置请参考 anytree 官方文档。