python绘制ast

可以使用Python自带的ast模块进行AST（Abstract Syntax Tree，抽象语法树）的解析和遍历，然后使用第三方库（如graphviz）生成AST的可视化图。

以下是一个简单的示例代码，可以生成一个函数的AST可视化图：

import ast
import graphviz

# 定义一个函数
def func(a, b):
    c = a + b
    d = a - b
    return c * d

# 解析函数的AST
tree = ast.parse(inspect.getsource(func))

# 定义AST节点和边的样式
node_attr = dict(style='filled', shape='box', align='left', fontsize='12', ranksep='0.1')
edge_attr = dict(arrowsize='0.7', fontsize='10', color='gray')

# 生成AST的可视化图
dot = graphviz.Digraph()
dot.attr('node', **node_attr)
dot.attr('edge', **edge_attr)

def add_nodes_edges(node):
    # 添加节点
    dot.node(str(node), str(node.__class__.__name__))
    # 遍历节点的子节点
    for child in ast.iter_child_nodes(node):
        # 添加边
        dot.edge(str(node), str(child))
        # 递归添加子节点和边
        add_nodes_edges(child)

add_nodes_edges(tree)

# 渲染并保存图像
dot.render('ast', view=True)

该代码会生成一个名为`ast`的图像文件，并自动打开文件进行预览。可以根据需要修改节点和边的样式，并添加更多节点和边的信息。