Advertisement
Guest User

Untitled

a guest
Sep 16th, 2019
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.63 KB | None | 0 0
  1. from anytree import AnyNode, RenderTree, PreOrderIter
  2. from pandas import DataFrame
  3.  
  4. root = AnyNode(id=1, name="Parent", children=[
  5. AnyNode(id=2, name="Child1", children=[
  6. AnyNode(id=4, name="GrandChild1"),
  7. AnyNode(id=5, name="GrandChild2"),
  8. ]),
  9. AnyNode(id=3, name="Child2", children=[
  10. AnyNode(id=6, name="GrandChild1"),
  11. AnyNode(id=7, name="GrandChild2"),
  12. ])
  13. ])
  14. print(RenderTree(root))
  15.  
  16. rows = []
  17. for node in PreOrderIter(root):
  18. rows.append((node.id, node.name, int(node.parent.id) if not node.is_root else 0))
  19.  
  20. dataFrame = DataFrame(rows, columns=("Id", "Name", "Parent"))
  21. print(dataFrame)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement