python———栏目排序

l = [
    {'category_name': '我是第一级', 'id': 8, 'parent_id': 0, 'update_time': None, 'create_time': None},
    {'category_name': '2222', 'id': 2, 'parent_id': 3, 'update_time': None, 'create_time': None},
    {'category_name': 'CCCC', 'id': 6, 'parent_id': 5, 'update_time': None, 'create_time': None},
    {'category_name': 'AAAA', 'id': 4, 'parent_id': 0, 'update_time': None, 'create_time': None},
    {'category_name': '3333', 'id': 1, 'parent_id': 2, 'update_time': None, 'create_time': None},
    {'category_name': '1111', 'id': 3, 'parent_id': 0, 'update_time': None, 'create_time': None},
    {'category_name': 'BBBB', 'id': 5, 'parent_id': 4, 'update_time': None, 'create_time': None},
    {'category_name': '我是第二级', 'id': 9, 'parent_id': 8, 'update_time': None, 'create_time': None},
    {'category_name': '我是第三级', 'id': 10, 'parent_id': 9, 'update_time': None, 'create_time': None},
    {'category_name': '我是第四级', 'id': 11, 'parent_id': 10, 'update_time': None, 'create_time': None},
]

l.insert(0, {'id': 0, 'name': 'root'}) # 插入根

for i in l:
    i['son'] = [] # item中插入son

index = {l[i]['id']: i for i in range(len(l))}  # 创建列表索引，每条记录id对应数据所在行数

for i in range(1, len(l)):
    parent_id = l[i]['parent_id']  # 找到父d
    l[index[parent_id]]['son'].append(l[i])  # 插入数据

print(l[0]['son']) # 打印数据


[{'category_name': 'AAAA', 'parent_id': 0, 'id': 4, 'update_time': None, 'create_time': None, 'son': [
    {'category_name': 'BBBB', 'parent_id': 4, 'id': 5, 'update_time': None, 'create_time': None, 'son': [
        {'category_name': 'CCCC', 'parent_id': 5, 'id': 6, 'update_time': None, 'create_time': None, 'son': []}]}]},
 {'category_name': '啊是打算', 'parent_id': 0, 'id': 3, 'update_time': None, 'create_time': None, 'son': []},
 {'category_name': '啊撒啊飒飒', 'parent_id': 0, 'id': 2, 'update_time': None, 'create_time': None, 'son': []},
 {'category_name': '测试一下', 'parent_id': 0, 'id': 1, 'update_time': None, 'create_time': None, 'son': []}]