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免费学习笔记（深入）”；

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

魔法映像企业网站管理系统

技术上面应用了三层结构，AJAX框架，URL重写等基础的开发。并用了动软的代码生成器及数据访问类，加进了一些自己用到的小功能，算是整理了一些自己的操作类。系统设计上面说不出用什么模式，大体设计是后台分两级分类，设置好一级之后，再设置二级并选择栏目类型，如内容，列表，上传文件，新窗口等。这样就可以生成无限多个二级分类，也就是网站栏目。对于扩展性来说，如果有新的需求可以直接加一个栏目类型并新加功能操作

下载

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

代码解析

• 函数 generate_tree 接收两个参数：data 是字典列表，parent_id 是父节点 id。
• 循环遍历字典列表，找到所有父节点 id 为 parent_id 的节点。
• 为每个节点创建一个字典，包含 id 和 name 键。
• 递归调用 generate_tree 函数生成子节点，并将子节点添加到 children 字段中。
• 将节点添加到 tree 中，并返回 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'}]}
        ]
    }
]

注意

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