有没有办法在 sympy 中剥离变量的所有函数的表达式？

Question

假设我有一个等式：

 eq = sym.Eq(5*x**2 + 2*x + 5*y + f_1(y) + f_2(x), 0)

Sympy 中有没有办法从上述表达式中去除 x 部分的 function 和 y 部分的 function ？

我的目标是这样的：

sym.strip(eq, x)
>>>5*x**2 + 2*x + f_2(x)
sym.strip(eq, y)
>>>5*y + f_1(y)

Answer 1

我不是 100% 确定这是否适用于您定义的方程，但这似乎对我来说适用于多项式。 试试这个或改编它。

def strip_sym(expr, sym):
    newExpr = 0
    for term in expr.args:
        newTerm = term if sym in term.free_symbols else 0
        newExpr = newExpr + newTerm
    return newExpr