Python中的树形数据结构表示方法

处理层级分明的数据，例如网站菜单或文件系统，需要合适的结构来清晰地表达父子关系。 Python 提供多种方法来实现树形结构，其中字典和自定义类是常用的选择。

方法一：使用嵌套字典

嵌套字典是一种简洁的方式来表示树形结构。 例如：

立即学习“Python免费学习笔记（深入）”；

Python Pep Authoring

依据官方 python/peps仓库规则起草、审阅、编辑或准备 Python 改进提案（PEP），用于处理 PEP 文本和元数据头部。

下载

tree_data = {
    'root': {
        'children': [
            {'name': 'A', 'children': [{'name': 'AA'}, {'name': 'AB'}]},
            {'name': 'B', 'children': [{'name': 'BA'}]}
        ]
    }
}

在这个例子中，'root' 是根节点，每个节点包含一个 'children' 键，其值为一个列表，列表中的每个元素都是一个子节点（字典）。 这种方法直观易懂，但对于复杂树形结构，遍历和操作可能会变得比较繁琐。

方法二：自定义类

自定义类可以提供更清晰的结构和更方便的操作方法。 例如：

class TreeNode:
    def __init__(self, name, children=None):
        self.name = name
        self.children = children if children is not None else []

root = TreeNode('root', [
    TreeNode('A', [TreeNode('AA'), TreeNode('AB')]),
    TreeNode('B', [TreeNode('BA')])
])

使用自定义类，可以更方便地添加方法来遍历、搜索和修改树形结构。 例如，可以添加一个方法来打印树形结构：

def print_tree(node, level=0):
    print("  " * level + node.name)
    for child in node.children:
        print_tree(child, level + 1)

print_tree(root)

选择哪种方法取决于具体需求和复杂程度。 对于简单的树形结构，嵌套字典可能足够；而对于复杂结构或需要更多操作的场景，自定义类则更具优势。