在 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, ordered dot = Function(``'dot'``) def dot_expr(a, b): ``# 步骤1:强制对称性 ------ 总按 canonical order 排序参数 ``A, B = ordered((a, b)) ``if A != a: ``return dot_expr(B, A) ``# 步骤2:利用线性性展开第一个参数(若为和式) ``if isinstance(a, Add): ``terms = a.as_ordered_terms() # 确保展开顺序稳定 ``return Add(*[dot_expr(term, b) ``for term in terms]) ``# 步骤3:基础情形 ------ 返回未计算的符号函数 ``return dot(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, a+b) 的展开逻辑。
更紧凑的等价实现(利用 Add.make_args 和笛卡尔积)如下:
|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| from sympy.utilities.iterables import cartes def dot_expand(a, b): ``args_a = Add.make_args(a) ``args_b = Add.make_args(b) ``return sum(dot(*ordered((ia, ib))) ``for ia, 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 控制流,而非寄望于黑盒简化器自动推导。