使用自定义符号简化sympy

Question

假设我有sympy变量x, y和z

我有表达式(x + y) (x + y) + x

是否有可能将z定义为x + y，以便可以将上述表达式“简化”为z^2 + x？

即，我想在简化表达式中尽可能使用z。

Answer 1

是，使用subs，如

>>> var('x:z')
(x, y, z)
>>> ((x + y)*(x + y) + x).subs(x+y,z)
x + z**2