我有一个带有自引用外键的 Django 模型,所以我的模型(作为最基本形式的类)看起来像:
class MyObj(object):
def __init__(self, id, ttl, pid):
self.id = id
self.name = ttl
self.parentid = pid
所以我的数据样本可能如下所示:
nodes = []
nodes.append(MyObj(1,'a',0))
nodes.append(MyObj(2,'b',0))
nodes.append(MyObj(3,'c',1))
nodes.append(MyObj(4,'d',1))
nodes.append(MyObj(5,'e',3))
nodes.append(MyObj(6,'f',2))
我已经到了可以将其转换为嵌套字典的地步:
{'a': {'c': {'e': {}}, 'd': {}}, 'b': {'f': {}}}
使用 Converting tree list to hierarchy dict作为指南,但我需要它的形式可以用于 Django 的 unordered_list 过滤器。
所以我的问题是,如何从(或者)嵌套字典获取嵌套列表/元组或直接从源数据获取嵌套列表?我似乎无法获得正确嵌套列表的递归函数(因为在列表中我无法按名称引用“子树”)
eval(string_rep_of_dictionary.replace(':',',').replace('{','[').replace('}',']')) 似乎只是让我到达那里,但这似乎一个糟糕的解决方案?
请您参考如下方法:
尝试
lists = {}
for n in nodes:
b = lists.setdefault(n.id, [])
lists.setdefault(n.parentid, []).extend([n.name, b])
print lists[0]
或者,使用 collections.defaultdict
lists = collections.defaultdict(list)
for n in nodes:
lists[n.parentid] += [n.name, lists[n.id]]
print lists[0]
两者都会打印
['a', ['c', ['e', []], 'd', []], 'b', ['f', []]]
编辑:为了去除空列表,第二次遍历节点:
for n in nodes:
if not lists[n.id]:
lists[n.parentid].remove(lists[n.id])
