
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 函数以递归的方式遍历数据,将父节点与其子节点关联,形成树形结构。具体流程如下:
- 遍历数据,查找当前节点id的子节点。
- 如果找到子节点,将其添加到当前节点的子节点列表中。
- 将当前节点添加到树中。
- 递归调用 build_tree 函数,以当前节点的id作为子节点的父节点id。
自定义适用场景
该函数让你灵活处理不同格式和层级的嵌套数据,适用于:
- 将菜单项转换为树形结构以生成导航栏。
- 将目录文件结构转换为树形结构以浏览文件系统。
- 对具有层次关系的数据进行高效操作。










