写一个树形图的代码

以下是一个使用Python语言绘制树形图的示例代码：

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.children = []

def print_tree(root, level=0):
    if not root:
        return
    print('  ' * level + '|__' + str(root.val))
    for child in root.children:
        print_tree(child, level + 1)

# 示例用法
root = TreeNode(1)
root.children = [TreeNode(2), TreeNode(3), TreeNode(4)]
root.children[0].children = [TreeNode(5), TreeNode(6)]
root.children[2].children = [TreeNode(7)]
print_tree(root)

输出结果：

|__1
  |__2
    |__5
    |__6
  |__3
    |__7
  |__4

该代码定义了一个`TreeNode`类，表示树的节点，节点包含一个值和子节点列表。`print_tree`函数是一个递归函数，用于打印树形图。它首先打印当前节点的值，然后递归地打印每个子节点，并在每个子节点前添加一些空格以形成树形结构。最后，我们创建一个示例树并打印出来，输出结果显示了树形结构。