Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from anytree import AnyNode, RenderTree, PreOrderIter
- from pandas import DataFrame
- root = AnyNode(id=1, name="Parent", children=[
- AnyNode(id=2, name="Child1", children=[
- AnyNode(id=4, name="GrandChild1"),
- AnyNode(id=5, name="GrandChild2"),
- ]),
- AnyNode(id=3, name="Child2", children=[
- AnyNode(id=6, name="GrandChild1"),
- AnyNode(id=7, name="GrandChild2"),
- ])
- ])
- print(RenderTree(root))
- rows = []
- for node in PreOrderIter(root):
- rows.append((node.id, node.name, int(node.parent.id) if not node.is_root else 0))
- dataFrame = DataFrame(rows, columns=("Id", "Name", "Parent"))
- print(dataFrame)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement