python - 伪环中 Sympy 中的自定义简化规则

Question

我正在使用 Sympy 来表示自定义交换伪环的元素。我想让 Sympy 了解我域中的一些特定简化规则，例如

x*x = x
n*x = x

有了这些规则，我希望 x*y*x + y*x 之类的东西在 x*y 中得到简化。有没有办法在 Sympy 中做到这一点？你推荐其他库来做吗？ (我只使用了expand、simplify、相等性测试、subs 和parse_expr)

Answer 1

经过几次实验，我最好的猜测是根据常量的扩展路径使用常量的自定义替换(参见 Sympy doc on Epath)

formula = expand(simplify(EPath("/*/*/Integer").apply(formula, lambda x: Integer(1))))
formula = expand(simplify(EPath("/*/Integer").apply(formula, lambda x: Integer(1))))
formula = expand(simplify(EPath("/Integer").apply(formula, lambda x: Integer(1))))

这三行基本上完成了所需的简化(例如 x*y*x + y*x -> x*y)。

可能有更优雅的解决方案，但我发布了这个答案，因为它可以帮助其他人。