确定 SymPy 表达式的各个项

Question

对于我正在从事的项目，我必须从任意 SymPy 表达式中选择特定术语。为此，我使用 .args 如果表达式包含多个不同的项，它可以正常工作，但如果仅存在单个项，则返回项中的各个系数。请参阅以下示例：

import sympy as sym

x, c0, c1 = sym.symbols('x c0 c1')

f0 = c0*x
f1 = c1*x**2
f2 = c0*x + c1*x**2

print(f0.args) # Output: (c0, x)    Desired: (c0*x)
print(f1.args) # Output: (c1, x**2) Desired: (c1*x**2)
print(f2.args) # Output: (c0*x, c1*x**2)

此方法的工作原理与我希望的 f2 一样，它返回 (c0*x, c1*x**2)。然而，对于f0和f1，它分别返回系数c0、c1和x。对于单项表达式，系数和 x 也作为乘法表达式返回，我将如何实现所需的输出？

请注意，理想情况下，我希望此方法适用于任何形式的表达式，并且任意依赖于 x 和任意数量的系数。

Answer 1

功劳归于奥斯卡·本杰明。由于我无法接受他们的评论作为答案，因此我将其重新发布到此处，同时还提供了一个具体的示例。

“您可以使用 Add.make_args(f0)”

这将如下所示：

import sympy as sym

x, c0, c1 = sym.symbols('x c0 c1')

f0 = c0*x
f1 = c1*x**2
f2 = c0*x + c1*x**2

print(sym.Add.make_args(f0))
print(sym.Add.make_args(f1))
print(sym.Add.make_args(f2))

它将返回：

(c0*x,)
(c1*x**2,)
(c0*x, c1*x**2)

Answer 2

您需要使用poly来说明您有多项式：

import sympy as sym
from sympy import symbols, Poly


x, c0, c1 = sym.symbols('x c0 c1')

f0 = sym.poly(c0*x)
f1 = sym.poly(c1*x**2)
f2 = c0*x + c1*x**2

print(f0.args) # Output: (c0, x)    Desired: (c0*x)
print(f1.args) # Output: (c1, x**2) Desired: (c1*x**2)
print(f2.args) # Output: (c0*x, c1*x**2)

返回

(c0*x, x, c0)
(c1*x**2, x, c1)
(c0*x, c1*x**2)

以获取单项式，只需使用

f0.args[0]

which给出

c0*x