无限极分类

dreams_dream

已于 2022-05-15 22:45:37 修改

阅读量115

点赞数

文章标签： flask python 后端 django

于 2022-05-15 22:45:08 首次发布

本文链接：https://blog.csdn.net/dreams_dream/article/details/124789896

版权

该博客展示了两种不同的方法，非递归和递归，来构建菜单树结构。通过Python代码实现，从给定的数据中组织成树形结构，其中非递归方法通过迭代查找父节点并附加子节点，而递归方法则是通过函数递归查找子节点。这两种方法都是为了将扁平化的数据转换为层次结构，适用于数据组织和显示。

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

非递归方式

import json

def format(data):
    print(json.dumps(data, indent=4, ensure_ascii=False, sort_keys=False, separators=(',', ':')))

def getTree(menu_list):
    # 处理数据
    menu_map = {}
    for item in menu_list:
        item["child"] = []
        menu_map[item["id"]] = item

    tree = []
    for item in menu_map.values():
        if menu_map.get(item["pid"]): # 找儿子
            menu_map[item["pid"]]["child"].append(item)
        else: # 找出所有的顶级
            tree.append(item)

    return tree


if __name__ == '__main__':
    menu_list = [
        {
            "id": 1,
            "pid": 0,
            "name": "top1"
        },
        {
            "id": 2,
            "pid": 0,
            "name": "top2"
        },
        {
            "id": 3,
            "pid": 1,
            "name": "top1-sub1"
        },
        {
            "id": 4,
            "pid": 1,
            "name": "top1-sub2"
        },
        {
            "id": 5,
            "pid": 3,
            "name": "top1-sub1-sub1"
        },
        {
            "id": 6,
            "pid": 2,
            "name": "top2-sub1"
        },
    ]

    format(getTree(menu_list))

递归方式


import json

def format(data):
    print(json.dumps(data, indent=4, ensure_ascii=False, sort_keys=False, separators=(',', ':')))

# 父亲找儿子
def getTree(menu_list,pid):

    tree = []
    for item in menu_list:
        if item["pid"] == pid:
            item["child"] = getTree(menu_list, item["id"]) # 找儿子的过程
            tree.append(item)
    return tree

if __name__ == '__main__':
    menu_list = [
        {
            "id": 1,
            "pid": 0,
            "name": "top1"
        },
        {
            "id": 2,
            "pid": 0,
            "name": "top2"
        },
        {
            "id": 3,
            "pid": 1,
            "name": "top1-sub1"
        },
        {
            "id": 4,
            "pid": 1,
            "name": "top1-sub2"
        },
        {
            "id": 5,
            "pid": 3,
            "name": "top1-sub1-sub1"
        },
        {
            "id": 6,
            "pid": 2,
            "name": "top2-sub1"
        },
    ]

    format(getTree(menu_list,0))

原文地址：https://www.cnblogs.com/sookie2020/p/14192879.html