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代码解析
- 函数 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'}]}
]
}
]注意
代码中未考虑环路情况。如果原始数据中存在环路，递归调用会陷入无限循环。实际应用中，需要对原始数据进行环路检测和处理。
