python 列表转树结构

文子阳

已于 2022-09-09 15:38:40 修改

阅读量1.5k

点赞数 1

分类专栏： python 文章标签：前端

于 2022-09-09 15:37:17 首次发布

本文链接：https://blog.csdn.net/wenxingchen/article/details/126783943

版权

python 专栏收录该内容

127 篇文章 11 订阅

订阅专栏

这是一个将普通列表转换为树结构列表的Python函数，通过关键字'id'、'parent_id'和'children'来构建树形数据。函数首先将输入列表转化为以'id'为键的字典，然后遍历列表，根据'parent_id'找到对应父节点并添加子节点，最终形成树结构。

摘要由CSDN通过智能技术生成

适用于普通列表转树结构列表。

"""
列表转树函数

关键字：id、parent_id、children
"""


def list2tree(data: list) -> list:
    # 转成ID为Key的字典
    mapping: dict = dict(zip([i['id'] for i in data], data))

    # 树容器
    container: list = []

    for d in data:
        # 如果找不到父级项，则是根节点
        parent: dict = mapping.get(d['parent_id'])
        if parent is None:
            container.append(d)
        else:
            children: list = parent.get('children')
            if not children:
                children = []
            children.append(d)
            parent.update({'children': children})
    return container


if __name__ == '__main__':
    data: list = [
        {'id': 1, 'parent_id': 0, 'name': '用户管理', 'url': 'https://www.baidu.com'},
        {'id': 2, 'parent_id': 0, 'name': '菜单管理', 'url': 'https://www.baidu.com'},
        {'id': 3, 'parent_id': 1, 'name': '新增用户', 'url': 'https://www.baidu.com'},
        {'id': 4, 'parent_id': 1, 'name': '删除用户', 'url': 'https://www.baidu.com'},
        {'id': 5, 'parent_id': 2, 'name': '新增菜单', 'url': 'https://www.baidu.com'},
        {'id': 6, 'parent_id': 2, 'name': '删除菜单', 'url': 'https://www.baidu.com'},
    ]

    # 打印验证一下
    for i in list2tree(data):
        print(i)