Read data from a file and create a tree using anytree in python

Question

Is there a way to read data from a file and construct a tree using anytree?

Parent Child
A      A1
A      A2
A2     A21

I can do it with static values as follows. However, I want to automate this by reading the data from a file with anytree.

>>> from anytree import Node, RenderTree
>>> A = Node("A")
>>> A1 = Node("A1", parent=A)
>>> A2 = Node("A2", parent=A)
>>> A21 = Node("A21", parent=A2)

Output is

A
├── A1
└── A2
    └── A21

Asked By: hshantanu

||

Source

Answer 1

This assumes that the entries are in such an order that a parent node was always introduced as a child of another node beforehand (root excluded).

With that in mind, we can then iterate over the lines, split them (I used split, regex would work too) and create the new nodes.

For how to get a reference to the parent by name, I came up with two solutions:

First, find the parent by name using 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))

Second, just cache them in a dict while we create them:

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

Output:

A
├── A1
└── A2
    └── A21

Answered By: Fabian N.

Answer 2

bigtree is a Python tree implementation that integrates with Python lists, dictionaries, and pandas DataFrame.

For this scenario, assuming your file is in a list of parent-child tuples

relation_list = [("A", "A1"), ("A", "A2"), ("A2", "A21")]

from bigtree import list_to_tree_by_relation, print_tree
root = list_to_tree_by_relation(relation_list)
print_tree(root)

This results in output,

A
|-- A1
`-- A2
    `-- A21

Source: Creator of bigtree 😉

Answered By: Kay Jan

Read data from a file and create a tree using anytree in python

Question:

Answers: