Python：如何利用字典列表生成无限级树结构？

百变鹏仔 5天前 #Python

文章标签字典

python：如何根据字典生成无限级树结构（子节点为 children）

问题描述

给定以下字典列表，如何生成一个无限级树结构，其中子节点的键为 children？

rows = [    {'pid': 0, 'id': 1, 'name': 'aaa'},    {'pid': 0, 'id': 2, 'name': 'aaa'},    {'pid': 0, 'id': 3, 'name': 'aaa'},    {'pid': 1, 'id': 4, 'name': 'aaa'},    {'pid': 4, 'id': 5, 'name': 'aaa'},    {'pid': 5, 'id': 6, 'name': 'aaa'},]

解答

立即学习“Python免费学习笔记（深入）”；

可以用递归实现无限级树结构的生成。下面提供一个示例代码：

def generate_tree(data, parent_id=none):    tree = []    for item in data:        if item['parent_id'] == parent_id:            node = {'id': item['id'], 'name': item['name']}            children = generate_tree(data, item['id'])            if children:                node['children'] = children            tree.append(node)    return tree

代码解析

示例用法

data = [    {'id': 1, 'name': 'a', 'parent_id': none},    {'id': 2, 'name': 'b', 'parent_id': 1},    {'id': 3, 'name': 'c', 'parent_id': 1},    {'id': 4, 'name': 'd', 'parent_id': 2},    {'id': 5, 'name': 'e', 'parent_id': 3},    {'id': 6, 'name': 'f', 'parent_id': 3},    {'id': 7, 'name': 'g', 'parent_id': 4},    {'id': 8, 'name': 'h', 'parent_id': 5},]tree = generate_tree(data)print(tree)

输出

[    {        'id': 1,        'name': 'A',        'children': [            {'id': 2, 'name': 'B', 'children': [{'id': 4, 'name': 'D', 'children': [{'id': 7, 'name': 'G'}]}]},            {'id': 3, 'name': 'C', 'children': [{'id': 5, 'name': 'E', 'children': [{'id': 8, 'name': 'H'}]}, {'id': 6, 'name': 'F'}]}        ]    }]

注意

代码中未考虑环路情况。如果原始数据中存在环路，递归调用会陷入无限循环。实际应用中，需要对原始数据进行环路检测和处理。

文章推荐

Python：如何利用字典列表生成无限级树结构？

Python - 字典中键的值求和

Python程序向列表中添加元素

Python程序将列表中的所有元素连接成一个字符串

学习起来哪个更容易，SQL还是Python？

Python - 使用切片获取最后K个列表项的总和