文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

python的简单四则运算语法树可视化怎么实现

2023-07-05 21:51

关注

本文小编为大家详细介绍“python的简单四则运算语法树可视化怎么实现”,内容详细,步骤清晰,细节处理妥当,希望这篇“python的简单四则运算语法树可视化怎么实现”文章能帮助大家解决疑惑,下面跟着小编的思路慢慢深入,一起来学习新知识吧。

简单四则运算语法树可视化

如果想要动手尝试一下,需要安装一下这个 python 可视化库。

python的简单四则运算语法树可视化怎么实现

解析 AST

Python 的 ast 库有一个 parse 方法,可以把传入的内容,解析成一个 AST。然后我们使用 ast.dump 将其导出并打印。
注意:indent 这个参数是 Python 3.9 以后才有的,如果版本低的话,可以去掉,只会影响最后输出的格式。

python的简单四则运算语法树可视化怎么实现

好了,就是这么简单。我们已经做到了,因为这个库的功能很强大,但是这里只是用到一点点而已。其实这里已经可以看出基本的结构了,不过我的目的是生成这棵树的 JSON 表示。我想要使用上面的 Python 绘图库把它绘制出来,它所支持的输入是 JSON,并且它的格式为:

{"name": "A","children": ["name": "B","children": []]}

粗糙的遍历方法

"""Python's AST利用 Python 的 ast 模块来解析表达式(简单的二元运算),然后通过遍历语法树来生成 JSON 表示,再使用 PYthon 的库来将其可视化。这个程序的目的是为了验证自己写的简易解析器是否正确。"""import astimport json# 操作类型和操作符映射的字典OPERATORS = {    ast.Add: "+",    ast.Sub: "-",    ast.Mult: "*",    ast.Div: "/"}def generate(tree: ast.Module):    """    generate expression AST's representation of JSON    """    if not tree:        raise Exception("Emtpy AST tree!")    if tree.__class__ == ast.Module:        print(json.dumps({            "name": "Expr",            "children": [DFS(tree.body[0].value)]  # type: ignore        }, indent=4))def DFS(node):    """    DFS AST    """    if not node:        return {}    if node.__class__ == ast.BinOp:        return {            "name": "BinOp",            "children": [                {                    "name": "left",                    "children": [                        DFS(node.left)                    ]                },                DFS(node.op),                {                    "name": "left",                    "children": [                        DFS(node.right)                    ]                }            ]        }    if node.__class__ == ast.Constant:        return {            "name": "NUMBER",            "children": [                {                    "name": str(node.value)  # Python 的绘图库,必须是字符串才能正常显示                }            ]        }    if node.__class__ in [ast.Add, ast.Sub, ast.Mult, ast.Div]:        return {            "name": "Op",            "children": [                {                    "name": OPERATORS[node.__class__]                }            ]        }    # 这里我只处理 加减乘除和数字类型的运行    raise Exception("There is not support extra type.")if __name__ == "__main__":    ast_tree = ast.parse("1+2+3+4+5")    print(ast.dump(ast_tree, indent=4))    generate(ast_tree)

运行结果:
我这里会输出两个东西,一个是 AST 的 dump;另一个是 AST 的 JSON 表示(逻辑结构的 JSON 表示,不是对象的 JSON 表示)。

python的简单四则运算语法树可视化怎么实现

渲染显示

把打印出来的 JSON 字符串复制进文件,命名为 data.json。我感觉直接输出到控制台蛮有意思的,我喜欢直接看到它的结果。
执行如下命令:pytm-cli -d TB -i data.json -o demo.html
在浏览器打开 demo.html 即可看到效果了。

python的简单四则运算语法树可视化怎么实现

主流的遍历方法

上面这种遍历方法虽然便于理解,但是难以扩展。AST 通常是通过 访问者模式 进行遍历的,而且 ast 库也提供了几种遍历方法。
因为这里只需要遍历来生成 JSON,并不需要修改AST本身,所以我们只看下面这两种即可。显然第一种是不能用的,原因已经用蓝色标记出来了。它自己说了如果你不关心上下文,因为生成 JSON 实际上是需要关注这个的。所以,我选择下面的 ast.NodeVisitor。使用它也很简单,继承这个类,然后对不同的节点写不同的处理逻辑就行了(这样把不同节点的逻辑分开了,降低了代码的耦合性)。

python的简单四则运算语法树可视化怎么实现

完整代码

"""Python's AST利用 Python 的 ast 模块来解析表达式(简单的二元运算),然后通过遍历语法树来生成 JSON 表示,再使用 PYthon 的库来将其可视化。这个程序的目的是为了验证自己写的简易解析器是否正确。"""import astimport json# 操作类型和操作符映射的字典OPERATORS = {    ast.Add: "+",    ast.Sub: "-",    ast.Mult: "*",    ast.Div: "/"}class JSONVisitor(ast.NodeVisitor):    """    JSON visitor: Traversal AST and generate JSON representation    """    def visit_Module(self, node):        module = {            "name": "Module",            "children": []        }        for sub_node in node.body:            module["children"].append(self.visit(sub_node))        return module    def visit_Expr(self, node):        return {            "name": "Expr",            "children": [                self.visit(node.value)            ]        }    def visit_BinOp(self, node):        return {            "name": "BinOp",            "children": [                {                    "name": "left",                    "children": [                        self.visit(node.left)                    ]                },                self.visit(node.op),                {                    "name": "right",                    "children": [                        self.visit(node.right)                    ]                }            ]        }    def visit_Constant(self, node):        return {            "name": "NUMBER",            "children": [{                "name": str(node.value)  # # Python 的绘图库,必须是字符串才能正常显示            }]        }    def visit_Add(self, node):        return self.__visit(node)    def visit_Sub(self, node):        return self.__visit(node)    def visit_Mult(self, node):        return self.__visit(node)    def visit_Div(self, node):        return self.__visit(node)    def __visit(self, node):        return {            "name": "Op",            "children": [{                "name": OPERATORS[node.__class__]            }]        }if __name__ == "__main__":    ast_tree = ast.parse("1+2+3+4+5")    visitor = JSONVisitor()    json_str = visitor.visit(ast_tree)    print(json.dumps(json_str, indent=4))

前面那个粗糙版本是直接从 Expr 开始的,这个优雅点的版本,我就把 Module 节点也添加进去了。

python的简单四则运算语法树可视化怎么实现

读到这里,这篇“python的简单四则运算语法树可视化怎么实现”文章已经介绍完毕,想要掌握这篇文章的知识点还需要大家自己动手实践使用过才能领会,如果想了解更多相关内容的文章,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯