如何打印使用类实现的树数据结构？

Question

我知道pprint可以漂亮地打印嵌套列表或字典，它们都是树结构的类型，但我想pprint一个基于类的树，其中包含任意数量的子节点，这样节点就会向右缩进，具体取决于它们树中的水平。

可以用pprint做我想做的事情，如果有，怎么做？ 我可以编写自己的函数来打印一棵树，但我想检查一下是否需要这样做。

请考虑以下示例：

class Tree:
    def __init__(self, value, *args):
        self.value = value
        self.children = args

tree = Tree("+", Tree(1), Tree("*", Tree(2), Tree(3)))

这是我的预期输出，如果我们要漂亮打印树：

+
    1
    *
        2
        3

再一次，我当然可以推出自己的功能，但我想看看我是否可以使用pprint如果这样会更容易，我只是忽略了一些东西。

pprint文档没有提供我的用例示例。

如果它有所作为，我使用的是Python 2.7。

Answer 1

pprint适用于已定义其__repr__方法的任意类。

为了得到你想要的东西，我认为你必须管理你自己的嵌套，此时你可以使用print 。

def __repr__(self, depth=1):
    return_string = [str(self.value)]
    for child in self.children:
        return_string.extend(["\n", " " * (depth+1), child.__repr__(depth+1)])
    return "".join(return_string)

然后pprint(tree)和print tree都给你

+
  1
  *
   2
   3