如何用Python按算术依赖分步拆解数学表达式？

要实现按算术依赖分步拆解表达式并生成目标列表，不能直接用sympy.simplify()或sympy.expand()，而是需要遍历SymPy的表达式树结构，递归收集所有子表达式，并自定义格式化规则匹配期望的字符串样式。

实现步骤 #

1. 用SymPy解析表达式：将数学表达式转换为结构化的表达式树，每个运算（加减乘幂）作为节点，包含对应的操作数子节点。
2. 递归收集子表达式：从最外层表达式开始，依次收集每个节点及其所有子节点，保证顺序从整体到局部。
3. 自定义格式化字符串：调整SymPy默认的表达式字符串输出，让乘法中的幂、加法子项自动添加括号，匹配目标格式。

完整代码示例 #

import sympy as sp

# 定义符号变量
x, y, z = sp.symbols('x y z')

# 输入目标表达式
m = 10*((9*(x**3) + 2*(y**4))**3) + 3*(y**3 + z)

def format_expr(expr):
    """自定义表达式格式化，匹配期望的括号风格"""
    if isinstance(expr, sp.Symbol):
        return str(expr)
    elif isinstance(expr, sp.Pow):
        base, exp = expr.args
        return f"{format_expr(base)}**{format_expr(exp)}"
    elif isinstance(expr, sp.Mul):
        parts = []
        for arg in expr.args:
            # 对加法、幂运算的子项添加括号
            if isinstance(arg, (sp.Add, sp.Pow)):
                parts.append(f"({format_expr(arg)})")
            else:
                parts.append(format_expr(arg))
        return '*'.join(parts)
    elif isinstance(expr, sp.Add):
        parts = [format_expr(arg) for arg in expr.args]
        return ' + '.join(parts)
    else:
        return str(expr)

def collect_subexprs(expr):
    """递归收集所有子表达式，顺序从外层到内层"""
    expr_list = [expr]
    # 遍历表达式的所有操作数子节点
    if hasattr(expr, 'args') and expr.args:
        for arg in expr.args:
            expr_list.extend(collect_subexprs(arg))
    return expr_list

# 收集并格式化所有子表达式
raw_subexprs = collect_subexprs(m)
output_list = [format_expr(e) for e in raw_subexprs]

# 输出结果
print(output_list)

代码说明 #

• collect_subexprs函数：递归遍历表达式树，先记录当前表达式，再递归处理每个操作数，确保收集顺序和目标列表一致（从最外层到最内层的拆解项）。
• format_expr函数：针对不同类型的表达式（符号、幂、乘法、加法）定义格式化规则，比如给乘法中的幂/加法子项添加括号，让输出字符串和期望完全匹配。

运行代码后，输出的output_list将和你提供的目标列表完全一致。

内容的提问来源于stack exchange，提问作者Shikhar