Python 有没有树形递归的快捷方法?

作者站长头像
站长
· 阅读数 5
  • python 有系统内置的包或者函数能做到么?
[
    {"id": 1, "name": "sql", "fid": 0},
    {"id": 2, "name": "sql", "fid": 1},
    {"id": 3, "name": "sql", "fid": 0},
    {"id": 4, "name": "sql", "fid": 3},
]
[
    {"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}]},
]
回复
1个回答
avatar
test
2024-06-28

内置函数我目前没见过,不过数据结构算法是可以实现的,写了一个你参考下:

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)

我大概解释一下,build_tree是一个递归函数,它接收一个数据列表和一个根节点ID作为参数。在每次递归调用中,它遍历数据列表,找到与当前节点ID匹配的子节点,并将其加入到当前节点的子节点列表中。递归终止条件是没有匹配的子节点。最终,函数返回树形结构。

通过调用build_tree函数将给定数据转换为树形结构,你可以按需修改代码以适应你的实际需求。

回复
likes
适合作为回答的
  • 经过验证的有效解决办法
  • 自己的经验指引,对解决问题有帮助
  • 遵循 Markdown 语法排版,代码语义正确
不该作为回答的
  • 询问内容细节或回复楼层
  • 与题目无关的内容
  • “赞”“顶”“同问”“看手册”“解决了没”等毫无意义的内容