在Python中垂直打印树吗？像一棵真正的树

Question

我正在尝试打印一棵树，问题是我找不到任何其他方法，只能像这样打印它：

1
|__2
   |__3
      |__33
   |__4
|__22

但有没有办法像这样打印它：

      1
     / \
    2  22
   / \
  3   4
  |
  33

这是我的代码：

class treeNode:

    def __init__(self,data):
        self.data = data
        self.children = []
        self.parent = None
        self.nextSibling = None
        self.level = 0
        self.prevSibling = None

在这里，我为树节点创建了一个类，我知道代码很糟糕，但我仍然是初学者，我将在未来改进它。我使用列表来存储子列表和链表来对兄弟姐妹进行排序，因为它稍后在尝试打印实际树时会很有用。

    def add_children(self,*child):
        for i in child:
            i.parent = self
            self.children.append(i)
            if len(self.children) > 1:
                self.children[-2].nextSibling = i

class Tree:

    def __init__(self,root:treeNode):
        self.root = root
        self.depth = 0

    def print_tree(self):
        print(self.root.data)
        kid = self.root.children[0]
        while (kid != self.root):
            kid.level_func()
            print("  " * (kid.level - 1) * 2 + "|" + "__",kid.data)
            if len(kid.children) == 0:
                while kid.parent:
                    if kid.nextSibling:
                        kid = kid.nextSibling
                        break
                    else:
                        kid = kid.parent
            else:
                kid = kid.children[0]

我打印树的方式是这样的：打印当前节点，如果当前节点有任何子节点，则转到子节点，如果没有，则转到当前节点的父节点的下一个兄弟节点，它将继续这样，直到到达根节点。

提前致谢 :)

Answer 1

如果您的树是二叉树，则可以在此处使用printBTree函数获得第二种形式

例如：

              80
          ___/  \___
        50          90
     __/  \__      /
   10        60  85
  /  \      /  \
 5    30  55    70
        \
         35

您可以尝试为具有多个子树的树创建一个变体，但很快就会耗尽水平空间。

对于这些树，缩进格式是可取的，尽管你可以让它看起来更好一点：

def printTree(node,nodeInfo,indent=""):
    label,children = nodeInfo(node)
    print(indent[:-3] + "|_ "*bool(indent) + str(label))
    for more,child in enumerate(children,1-len(children)):
        childIndent = "|  " if more else "   "
        printTree(child,nodeInfo,indent+childIndent)

用法:

T = {'data': 1, 'children':
        [ { 'data':2, 'children':
            [ {'data':6 }, {'data':7} ]},
          {'data':3},
          {'data':4},
          {'data':5, 'children':
           [{'data':10}]}
        ]
     }

printTree(T,lambda d:(d['data'],d.get('children',[])))

1
|_ 2
|  |_ 6
|  |_ 7
|_ 3
|_ 4
|_ 5
   |_ 10

您可以将 printTree 与 treeNode 类一起使用，如下所示：

printTree(myTree.root,lambda n:(n.data,n.children))