python用列表构建树_从Python中的邻接列表构建菜单树

假设你有这样的事情：

data = [

{ 'id': 1, 'parent_id': 2, 'name': "Node1" },

{ 'id': 2, 'parent_id': 5, 'name': "Node2" },

{ 'id': 3, 'parent_id': 0, 'name': "Node3" },

{ 'id': 4, 'parent_id': 5, 'name': "Node4" },

{ 'id': 5, 'parent_id': 0, 'name': "Node5" },

{ 'id': 6, 'parent_id': 3, 'name': "Node6" },

{ 'id': 7, 'parent_id': 3, 'name': "Node7" },

{ 'id': 8, 'parent_id': 0, 'name': "Node8" },

{ 'id': 9, 'parent_id': 1, 'name': "Node9" }

]

在列表中该功能迭代并创建树，收集每个节点的孩子是sub列表：

def list_to_tree(data):

out = {

0: { 'id': 0, 'parent_id': 0, 'name': "Root node", 'sub': [] }

}

for p in data:

out.setdefault(p['parent_id'], { 'sub': [] })

out.setdefault(p['id'], { 'sub': [] })

out[p['id']].update(p)

out[p['parent_id']]['sub'].append(out[p['id']])

return out[0]

例如：

tree = list_to_tree(data)

import pprint

pprint.pprint(tree)

如果父ID是没有的(不是0的)，修改这样的功能：

def list_to_tree(data):

out = {

'root': { 'id': 0, 'parent_id': 0, 'name': "Root node", 'sub': [] }

}

for p in data:

pid = p['parent_id'] or 'root'

out.setdefault(pid, { 'sub': [] })

out.setdefault(p['id'], { 'sub': [] })

out[p['id']].update(p)

out[pid]['sub'].append(out[p['id']])

return out['root']

# or return out['root']['sub'] to return the list of root nodes