如何在 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, 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 控制流,而非寄望于黑盒简化器自动推导。

相关推荐
牡丹雅忻110 分钟前
AES 加密模式演进:从 ECB、CBC 到 GCM 的 C# 深度实践
java·开发语言·c#
连续讨伐15 分钟前
php小结
开发语言·php
进击的程序猿~25 分钟前
Go 并发底层原理面试学习指南
开发语言·面试·golang
Python私教29 分钟前
Django 6.1 RC1 实测:FETCH_PEERS 两条 SQL 解决 N+1,select_related 还需要吗?
后端·python·django
脚踏实地皮皮晨30 分钟前
002002002_DepandencyObject类2
开发语言·windows·算法·c#·visual studio
玉鸯1 小时前
Agent 的任务编排:从 System Prompt 到 Hierarchical Multi-Agent
python·llm·agent
Python私教1 小时前
Django 6.0 自带 Tasks 到底能不能替代 Celery?跑完 3 组后台任务后我有答案了
后端·python·django
练习时长一年1 小时前
@SneakyThrows注解
开发语言
我的xiaodoujiao1 小时前
快速学习Python基础知识详细图文教程14--模块
开发语言·python·学习·测试工具
残影飞雪1 小时前
Ollama对话脚本
python