如何优化Python数学公式计算程序并实现分步求解功能？

原代码存在两个核心问题：一是调用formulate方法时会原地修改类属性self.formula，导致实例只能单次使用，属于逻辑缺陷；二是通过字符串替换修改变量值，容易出现变量名冲突（比如变量a和aa会误替换），且多次字符串遍历替换效率极低。
优化思路如下：

• 保留原始公式不修改，实现实例可重复调用
• 废弃字符串替换方案，改用eval的命名空间参数传递变量，既避免变量名冲突，又省去多次字符串操作的开销
• 用字典推导式快速解析输入的变量键值对，提升解析效率
• 避免使用sum这类Python内置关键字作为变量名，防止命名空间污染

优化后代码示例：

class Formula:
    def __init__(self, formula):
        self.raw_formula = formula  # 保留原始公式不修改
    
    def formulate(self, values):
        # 解析变量为字典
        var_dict = dict(pair.strip().split('=') for pair in values.split(';'))
        # 直接传变量上下文给eval，不修改原公式
        return eval(self.raw_formula, {}, var_dict)

# 测试用例
calc = Formula('(a+b)/c')
print(calc.formulate('a=2;b=56;c=5'))  # 输出11.6
# 可重复调用
print(calc.formulate('a=10;b=20;c=6')) # 输出5.0

可以通过解析公式的抽象语法树（AST）实现分步计算，不需要手动处理运算优先级，实现逻辑清晰：

1. 用Python内置的ast模块把原始公式解析为AST树，树的叶子节点是变量/常量，非叶子节点是运算操作
2. 后序遍历AST树，从最内层的子节点开始计算，每完成一次运算就记录当前步骤的运算表达式、操作数、计算结果
3. 所有子节点计算完成后，再计算父节点的运算，直到根节点计算完成得到最终结果
4. 按运算顺序输出所有记录的步骤即可得到完整的分步求解过程

简易实现示例：

import ast

class StepFormula(Formula):
    def calculate_with_steps(self, values):
        var_dict = dict(pair.strip().split('=') for pair in values.split(';'))
        # 把变量值转成浮点型
        var_dict = {k: float(v) for k, v in var_dict.items()}
        self.steps = []
        self.step_count = 1

        # 递归计算AST节点
        def eval_node(node):
            if isinstance(node, ast.Name):
                return var_dict[node.id]
            if isinstance(node, ast.Constant):
                return node.value
            if isinstance(node, ast.BinOp):
                left_val = eval_node(node.left)
                right_val = eval_node(node.right)
                # 匹配运算符
                op_map = {
                    ast.Add: '+', ast.Sub: '-', ast.Mult: '*', ast.Div: '/'
                }
                op_symbol = op_map[type(node.op)]
                result = eval(f'{left_val}{op_symbol}{right_val}')
                # 记录步骤
                self.steps.append(f"步骤{self.step_count}：{left_val} {op_symbol} {right_val} = {result}")
                self.step_count += 1
                return result
        
        # 解析公式为AST
        tree = ast.parse(self.raw_formula, mode='eval').body
        final_result = eval_node(tree)
        # 输出步骤和结果
        for step in self.steps:
            print(step)
        print(f"最终结果：{final_result}")
        return final_result

# 测试
step_calc = StepFormula('(a+b)/c')
step_calc.calculate_with_steps('a=2;b=56;c=5')

运行后输出：

步骤1：2.0 + 56.0 = 58.0
步骤2：58.0 / 5.0 = 11.6
最终结果：11.6

内容的提问来源于stack exchange，提问作者Tara Nath Niraula