有没有办法从文件中读取数据并使用anytree构造树？
Parent Child
A A1
A A2
A2 A21
我可以使用静态值来做到这一点，如下所示。但是，我想通过从具有anytree的文件中读取数据来使其自动化。
>>> from anytree import Node, RenderTree
>>> A = Node("A")
>>> A1 = Node("A1", parent=A)
>>> A2 = Node("A2", parent=A)
>>> A21 = Node("A21", parent=A2)
输出为
A
├── A1
└── A2
└── A21
这假定条目的顺序是始终始终将父节点作为另一个节点的子节点引入（不包括根）。
考虑到这一点，我们然后可以遍历各行，拆分它们（我使用过split，regex也可以）并创建新节点。
对于如何通过名称引用父对象，我提出了两种解决方案：
首先，使用anytrees通过名称查找父对象find_by_attr
from anytree import Node, RenderTree, find_by_attr
with open('input.txt', 'r') as f:
lines = f.readlines()[1:]
root = Node(lines[0].split(" ")[0])
for line in lines:
line = line.split(" ")
Node("".join(line[1:]).strip(), parent=find_by_attr(root, line[0]))
for pre, _, node in RenderTree(root):
print("%s%s" % (pre, node.name))
其次，在创建它们时将它们缓存在字典中：
from anytree import Node, RenderTree, find_by_attr
with open('input.txt', 'r') as f:
lines = f.readlines()[1:]
root = Node(lines[0].split(" ")[0])
nodes = {}
nodes[root.name] = root
for line in lines:
line = line.split(" ")
name = "".join(line[1:]).strip()
nodes[name] = Node(name, parent=nodes[line[0]])
for pre, _, node in RenderTree(root):
print("%s%s" % (pre, node.name))
input.txt
Parent Child
A A1
A A2
A2 A21
输出：
A
├── A1
└── A2
└── A21
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句