python-列表嵌套字典，根据父子关系，id及parent_id，格式化为多层嵌套的格式

昵称真的不好想

已于 2022-07-14 13:58:48 修改

阅读量1k

点赞数 1

文章标签： python 数据结构

于 2022-07-14 11:52:25 首次发布

本文链接：https://blog.csdn.net/weixin_42013006/article/details/125781396

版权

层级列表：
menus = [
    {'id': 1, 'name': 'a一级', 'parent_id': 0}, {'id': 2, 'name': 'b一级', 'parent_id': 0},
    {'id': 3, 'name': 'a二级', 'parent_id': 1}, {'id': 4, 'name': 'a二级级', 'parent_id': 1},
    {'id': 5, 'name': 'b二级', 'parent_id': 2}, {'id': 6, 'name': 'b二级级', 'parent_id': 2},
    {'id': 7, 'name': 'a三级', 'parent_id': 3}, {'id': 8, 'name': 'b三级级', 'parent_id': 5},
]


def format_menu(menus):
    nodes = {}
    data_temp = []
    for item in menus:
        id, parentId, name = item["id"], item["parent_id"], item["name"]
        # nodes[]保存需要的字典格式
        nodes[id] = {'id': id, "parentId": parentId, "name": name, 'children': []}
        # data_temp 保存id,parentId
        data_temp.append({'id': id, "parentId": parentId})
    data = []
    for i in data_temp:
        id = i['id']
        parent_id = i['parentId']
        node = nodes[id]
        if id == parent_id:
            data.append(node)
        else:
            parent = nodes.get(parent_id)
            if parent:
                parent['children'].append(node)
            else:
                data.append(node)
    print(data)

输出结果：

[
    {
        "id": 1,
        "parentId": 0,
        "name": "a一级",
        "children": [
            {
                "id": 3,
                "parentId": 1,
                "name": "a二级",
                "children": [
                    {
                        "id": 7,
                        "parentId": 3,
                        "name": "a三级",
                        "children": []
                    }
                ]
            },
            {
                "id": 4,
                "parentId": 1,
                "name": "a二级级",
                "children": []
            }
        ]
    },
    {
        "id": 2,
        "parentId": 0,
        "name": "b一级",
        "children": [
            {
                "id": 5,
                "parentId": 2,
                "name": "b二级",
                "children": [
                    {
                        "id": 8,
                        "parentId": 5,
                        "name": "b三级级",
                        "children": []
                    }
                ]
            },
            {
                "id": 6,
                "parentId": 2,
                "name": "b二级级",
                "children": []
            }
        ]
    }
]

借鉴地址：python 父子节点生成字典 - 蜗牛也是妞 - 博客园