PHP前端开发

Python树形递归如何便捷实现:需要自定义函数吗?

百变鹏仔 2个月前 (01-14) #Python
文章标签 递归

python 树形递归的便捷实现

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

系统自带的包或函数?

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

立即学习“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。

自定义适用场景

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