实现无限级分类的数据结构

关于多重分类的问题

分类下有子分类，子分类下又有子子分类要怎么规划这个结构

数据准备：

data=[
    {"cat_id":3,"name":"沙河","parent_id":1},
    {"cat_id":4,"name":"sb镇","parent_id":3},
    {"cat_id": 1, "name": "北京", "parent_id": 0},
    {"cat_id":5,"name":"昌平","parent_id":1},
    {"cat_id":6,"name":"青浦","parent_id":2},
    {"cat_id": 2, "name": "上海", "parent_id": 0},
]

期望获得结构形式：

北京
-沙河
--sb镇
-昌平
上海
-青浦

以上，北京是一个一级分类，沙河是北京分类的子分类，属于二级分类，sb镇是沙河镇的子分类，属于三级分类，然后昌平也是北京的子类，属于二级分类。 接下来是上海，属于一级分类，青浦是上海的子分类，属于二级分类。

要实现这样的结构，需要用递归

lis = []
def get_son1(data,parent_id=0,level=0,is_clear=True):
    #第一次递归的时候一定是能进去的
    if is_clear:
        lis.clear()
    # 循环遍历这个data列表，每个元素都是字典
    for item in data:
        #判断当前元素的父分类是不是等于传进来的参数parent_id
        if item['parent_id']==parent_id:
            #是的话就把传进来的level赋值给他的level，比如到这里，北京的level就是0了、
            item['level']=level
            #然后把北京推进这个空列表里。
            lis.append(item)
            #然后开始进入递归，把当前的id当做parent_id传进去
 			get_son1(data,parent_id=item["cat_id"],level=level+1,
 			is_clear=False)
    return lis

这是不用level的情况

实际使用情况，只会传过来一个分类和他的子分类，比如

    [{"cat_id":3,"name":"沙河","parent_id":1},
    {"cat_id":4,"name":"sb镇","parent_id":3},
    {"cat_id": 1, "name": "北京", "parent_id": 0},
    {"cat_id":5,"name":"昌平","parent_id":1}]

应用场景是，点击了一个分类，展示所有这个分类和子分类下的产品。

调用这个方法的时候，以下

 #获取所有分类中属于param.get("category_id")的子类，并包含它自己
    #data是所有分类，第二个参数是你点击的分类的id
all_cat_id = func.get_son_id_list(data,param.get("category_id"))

res_id=[]
#parent_id 就是你点击的分类的id，找这个分类下的所有子分类。
def get_son_id_list(data,parent_id,is_clear=True):
    #第一次走的时候就会进入这个if
    if is_clear:
        res_id.clear()
        #如果parent_id不等与0,就把自己加进去，因为等于0的时候，下面的逻辑会加
        if parent_id:
            res_id.append(parent_id)
    for item in data:
        #如果该分类的父级id为传过的parent_id，就把它的cat_id，存到res_id，接着找它自己有没有儿子，进行递归，是0的时候就代表是一个一级分类，也能满足，直接加进去。
        if item['parent_id']==parent_id:
            res_id.append(item['cat_id'])
            #开始递归，data还是原来的data，parent改成当前的分类id，找子子孙孙
            get_son_id_list(data,parent_id=item["cat_id"],
            is_clear=False)
    return res_id
#最后返回的是一个存放id的列表，根据这个列表里面的id，去查找在这些分类下的商品

面试题

不用递归的方式完成（推荐）：

def get_tree(data):
    #先定义一个空列表和空字典
    lists=[]
    tree={}
    #循环传进来的列表套字典
    for item in data:
        #把所有的分类的id和分类字典以键值对的形式存进这个空字典
        tree[item['cat_id']]=item
    #再次开始循环这个列表
    for i in data:
        #如果当前的分类的父分类是0
        if not i['parent_id']:
            #就直接把这个分类的id存进空列表
            lists.append(tree[i['cat_id']])
        else:
            #如果不是0的话，就把他的父分类的id复制给parent_id
            parent_id=i['parent_id']
            #判断这个分类的父有没有children这个键，没有的话就给他一个。
            if "children" not in tree[parent_id]:
                tree[parent_id]["children"]=[]
            #然后把当前分类的id，存到他的父分类的children里面。
            tree[parent_id]['children'].append(tree[i['cat_id']])

    return lists