python 树形递归的便捷实现

需要将嵌套结构的数据转换为树形结构吗？python 可以提供方便的方法。

系统自带的包或函数？

python 内置库中尚未提供直接处理树形递归的包或函数。

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

自定义数据结构算法

但我们可以利用 python 的数据结构和算法实现一个漂亮的解决方案：

Shadows Python Sensei

Python 最佳实践助手——代码规范、设计模式、性能优化、测试与类型注解。适用于编写或审查 Python 代码。

下载

def build_tree(data, root_id=0):
    tree = []
    for item in data:
        if item['fid'] == root_id:
            children = build_tree(data, item['id'])
            if children:
                item['children'] = children
            tree.append(item)
    return tree

使用方法

使用上述函数将嵌套数据转换为树形结构：

data = [
    {"id": 1, "name": "sql", "fid": 0},
    {"id": 2, "name": "sql", "fid": 1},
    {"id": 3, "name": "sql", "fid": 0},
    {"id": 4, "name": "sql", "fid": 3}
]

tree = build_tree(data)
print(tree)

输出：

[
    {"id": 1, "name": "sql", "fid": 0, "children": [{"id": 2, "name": "sql", "fid": 1}]},
    {"id": 3, "name": "sql", "fid": 0, "children": [{"id": 4, "name": "sql", "fid": 3}]}
]

原理说明

build_tree 函数以递归的方式遍历数据，将父节点与其子节点关联，形成树形结构。具体流程如下：

1. 遍历数据，查找当前节点id的子节点。
2. 如果找到子节点，将其添加到当前节点的子节点列表中。
3. 将当前节点添加到树中。
4. 递归调用 build_tree 函数，以当前节点的id作为子节点的父节点id。

自定义适用场景

该函数让你灵活处理不同格式和层级的嵌套数据，适用于：

• 将菜单项转换为树形结构以生成导航栏。
• 将目录文件结构转换为树形结构以浏览文件系统。
• 对具有层次关系的数据进行高效操作。