需要将嵌套结构的数据转换为树形结构吗?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。
自定义适用场景
该函数让你灵活处理不同格式和层级的嵌套数据,适用于:
- 将菜单项转换为树形结构以生成导航栏。
- 将目录文件结构转换为树形结构以浏览文件系统。
- 对具有层次关系的数据进行高效操作。