如何在 SymPy 中自定义点积函数并实现自动展开与对称化简化

在 SymPy 中原生 Function 类支持符号化定义但默认不具备代数运算规则。要让 dot(a b, c) 自动展开为 dot(a, c) dot(b, c)且保证 dot(a, b) 恒等于 dot(b, a)关键在于重载 Python 函数逻辑而非仅声明符号规则如 Eq 对象因为 SymPy 的 simplify() 或 expand() 不会自动应用用户定义的等式规则除非显式集成到重写系统中。以下是一种简洁、可靠且符合 SymPy 编程范式的实现方式from sympy import Function, symbols, Add, ordereddot Function(dot)def dot_expr(a, b):# 步骤1强制对称性 —— 总按 canonical order 排序参数A, B ordered((a, b))ifA ! a:returndot_expr(B, A)# 步骤2利用线性性展开第一个参数若为和式ifisinstance(a, Add):terms a.as_ordered_terms() # 确保展开顺序稳定returnAdd(*[dot_expr(term, b)forterm in terms])# 步骤3基础情形 —— 返回未计算的符号函数returndot(A, B)使用示例x, y, z symbols(x y z)print(dot_expr(x y, z)) # → dot(x, z) dot(y, z)print(dot_expr(x, y z)) # → dot(x, y) dot(x, z) 因对称性x 仍为第一参数print(dot_expr(x 1, y 2)) # → dot(1, 2) dot(1, y) dot(2, x) dot(x, y)✅注意上述实现隐含了 dot 关于两个变量均线性即双线性但仅通过对称性单侧展开达成若需严格控制线性方向如仅左线性应移除 ordered 调用并单独处理 dot(c, ab) 的展开逻辑。更紧凑的等价实现利用 Add.make_args 和笛卡尔积如下from sympy.utilities.iterables import cartesdef dot_expand(a, b):args_a Add.make_args(a)args_b Add.make_args(b)returnsum(dot(*ordered((ia, ib)))foria, ib in cartes(args_a, args_b))该版本将 dot(a, b) 视为所有单项式对的“张量积求和”天然满足双线性与对称性且代码更简明、不易出错。⚠️重要限制说明SymPy 原生 simplify()、expand()不会自动调用你的 dot_expr 函数你必须显式用 dot_expr(expr1, expr2) 构造表达式或封装为自定义 Dot 类并重写 _eval_expand 方法若需深度集成如在 expr.simplify() 中自动触发推荐继承 Function 并重写 _eval_expand_basic 和 _eval_simplify但复杂度显著上升替代方案Mathematica 可通过 UpValues 直接定义 dot 的代数规则Maple 支持 define 命令而 Python 生态中SymPy 仍是符号代数最成熟的选择辅以少量定制即可满足绝大多数需求。综上不依赖外部库、仅用 SymPy 原生机制通过函数式递归构造即可高效实现自定义点积的展开与规范化——核心在于将代数规则转化为可控的 Python 控制流而非寄望于黑盒简化器自动推导。