登录
首页 >  文章 >  python教程

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

时间：2024-12-02 16:54:34 100浏览 收藏

一分耕耘，一分收获！既然都打开这篇《Python：如何利用字典列表生成无限级树结构？》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新文章相关的内容，希望对大家都有所帮助！

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

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

问题描述

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

1
2
3
4
5
6
7
8
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'},
]

解答

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

1
2
3
4
5
6
7
8
9
10
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。

示例用法

1
2
3
4
5
6
7
8
9
10
11
12
13
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)

输出

1
2
3
4
5
6
7
8
9
10
[
    {
        '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：如何利用字典列表生成无限级树结构？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>