python - 从给定的字典创建一棵树

Question

我有一个 python 字典，我想用它创建一棵树。
字典是这样的:

dict_={"2":{'parent': "1"},"1":{'parent': None},"3":{'parent': "2"}}

在这种情况下，根是“1”

我尝试使用 treelib 库，但是当我在字典上迭代并创建节点时出现问题，但尚未创建其父节点。例如，如果我想为“2”创建一个节点，它的父节点(“1”)还没有创建，所以不能这样做。任何的想法？

Answer 1

您可以使用 treelib 执行以下操作:

from treelib import Node, Tree

dict_ = {"2": {'parent': "1"}, "1": {'parent': None}, "3": {'parent': "2"}}

added = set()
tree = Tree()
while dict_:

    for key, value in dict_.items():
        if value['parent'] in added:
            tree.create_node(key, key, parent=value['parent'])
            added.add(key)
            dict_.pop(key)
            break
        elif value['parent'] is None:
            tree.create_node(key, key)
            added.add(key)
            dict_.pop(key)
            break

tree.show()

输出

1
└── 2
    └── 3

这个想法是仅当父节点存在于树中或父节点为 None 时才添加节点。 .当父级为 None将其添加为 root。