Question

我正在生成符号函数，并使用SymPy对其进行简化。现在，我想一种“简化”符号的方法，这些符号代表尚未安装的函数中的常量参数。例如，如果我生成多项式，则可能生成这样的字符串

C*x*x + C*x + C*x + C,

SymPy将变成哪个

C*x^2 + 2*C*x + C.

现在，我想找到一种方法来获取此信息：

C*x^2 + C*x + C.

换句话说，有一种方法可以告诉SymPy某个符号是常量且未定义，因此

C+C -> C, C*C -> C, etc. Or more generally: f(C) = C, for any f(C)?

我的第一个想法是，也许有一个描述此属性的假设（例如Q.positive），我也许可以使用优化。但是，似乎并非如此。如果没有别的，我敢肯定有一种方法可以使用preorder_traversal来完成我想做的事情，但是我无法提出一种实现它的策略。任何帮助或想法都将受到赞赏。

Answer 1

也许是这样（适用于已完全扩展的表达式）：

def consim(eq, *v):
    con = numbered_symbols('c', cls=Dummy)
    reps = {}
    for i in preorder_traversal(eq):
        if i.is_Mul or i.is_Add:
            c, d = i.as_independent(*v)
            if c != i.identity and c.free_symbols:
                c = reps.setdefault(c, next(con))
    return eq.subs(reps)


>>> from sympy.abc import a, b, c, d, x
>>> eq = 2*a*x**2 + b*c*x + d + e
>>> consim(eq, x)
         2
c₀ + c₁⋅x  + c₂⋅x

您可能希望编号符号，并非所有符号都相同。

有没有办法在SymPy中处理常量函数参数？

1 个答案: