数据准备
city:[
{'cat_id':3,'name':'河北省','parent_id':0},
{'cat_id':4,'name':'邯郸','parent_id':1},
{'cat_id':7,'name':'魏县','parent_id':2},
{'cat_id':2,'name':'保定','parent_id':1},
{'cat_id':5,'name':'河南省','parent_id':0},
{'cat_id':9,'name':'安阳','parent_id':1},
],
需求
获取以下分类数据结构
'''
河北省
-邯郸
--魏县
-保定
河南省
-安阳
'''
河北省为一级分类,parent_id为0,邯郸是河北省的子分类,魏县是邯郸的子分类,以此类推
实现
方式一:递归实现
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