python 列表转树结构

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

"""
列表转树函数

关键字：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)