我有这个数学表达式：
tree = ast.parse('1 + 2 + 3 + x')
对应于以下抽象语法树：
Module(body=[Expr(value=BinOp(left=BinOp(left=BinOp(left=Num(n=1), op=Add(), right=Num(n=2)), op=Add(), right=Num(n=3)), op=Add(), right=Name(id='x', ctx=Load())))])
我想简化一下-也就是说，
Module(body=[Expr(value=BinOp(left=Num(n=6), op=Add(), right=Name(id='x', ctx=Load())))])
根据文档，我应该使用NodeTransformer类。文档中的建议如下：
请记住，如果您正在操作的节点具有子节点，则必须自己转换子节点或首先为该节点调用generic_visit（）方法。
我尝试实现自己的转换器：
class Evaluator(ast.NodeTransformer):
def visit_BinOp(self, node):
print('Evaluating ', ast.dump(node))
for child in ast.iter_child_nodes(node):
self.visit(child)
if type(node.left) == ast.Num and type(node.right) == ast.Num:
print(ast.literal_eval(node))
return ast.copy_location(ast.Subscript(value=ast.literal_eval(node)), node)
else:
return node
在这种特定情况下，它应该做的是将1 + 2简化为3，然后将3 + 3简化为6。它确实简化了我想简化的二进制运算，但是它没有更新原始的语法树。我尝试了不同的方法，但是仍然不知道如何以深度优先的方式递归简化所有二进制操作。有人能指出我正确的方向吗？
谢谢。
该方法有三个可能的返回值visit_*：
None 这意味着该节点将被删除，node （节点本身），这意味着将不会应用任何更改，因此，当您要用替换BinOpa时，Num您需要返回一个新Num节点。表达式的求值不能通过来完成，ast.literal_eval因为此函数仅求值文字（而不是任意表达式）。相反，您可以使用eval例如。
因此，您可以使用以下节点转换器类：
import ast
class Evaluator(ast.NodeTransformer):
ops = {
ast.Add: '+',
ast.Sub: '-',
ast.Mult: '*',
ast.Div: '/',
# define more here
}
def visit_BinOp(self, node):
self.generic_visit(node)
if isinstance(node.left, ast.Num) and isinstance(node.right, ast.Num):
# On Python <= 3.6 you can use ast.literal_eval.
# value = ast.literal_eval(node)
value = eval(f'{node.left.n} {self.ops[type(node.op)]} {node.right.n}')
return ast.Num(n=value)
return node
tree = ast.parse('1 + 2 + 3 + x')
tree = ast.fix_missing_locations(Evaluator().visit(tree))
print(ast.dump(tree))
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句