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

相关推荐
老歌老听老掉牙1 小时前
空间曲线数据的拼接与几何变换分析
python·曲线
Cc.Y2 小时前
Java零基础入门:字符串深度掌握——从基础API到StringBuilder性能优化
java·开发语言·性能优化
databook5 小时前
使用 SciPy 库进行时间序列分析
python·数据分析·scipy
java资料站5 小时前
十一、评估测试
开发语言·人工智能·python
szial6 小时前
网络爬虫与 CDP 实战(二):浏览器能访问,Python 却被拒绝?拆开登录态与 CSRF
爬虫·python·csrf
左手の明天6 小时前
SLIP 协议封装与解析详解:原理、C 代码与实战案例
linux·c语言·开发语言·c++
海上小飞龙6 小时前
【KMP算法-下篇】同一道题:Java 库函数 2600 微秒,手写 KMP 16 微秒
java·开发语言·算法
青少儿编程课堂7 小时前
后缀自动机解析:本质不同子串与最长重复片段统计
c++·python·算法·bfs·信息学竞赛
朝朝辞暮i7 小时前
C++ 第 12 课:局部变量、作用域、变量生命周期
开发语言·c++·算法
狗凯之家源码网7 小时前
淘客口令落地页生成器 PHP 版深度评测
开发语言·php