在sympy中隔离多元多项式的一个系数的最佳方法

Question

我有一个多元多项式（在一般情况下有许多变量），其系数列出了我需要读取的一些数据，但似乎sympy没有好的方法可以实现这一点。

collect 函数似乎是正确的想法，但当您使用它处理多个变量时，它实际上不会给您单独的单项式，而是给出依赖于您列出变量的顺序的奇怪单项式组合。

有人知道如何解决这个问题吗？

- xanderflood

Answer 1

多项式模块的文档列出了许多处理系数的方法。例如：

>>> import sympy
>>> x,y,z = sympy.symbols('x,y,z')
>>> p = sympy.poly((x+2*y-z)**3)
>>> p.coeffs()
[1, 6, -3, 12, -12, 3, 8, -12, 6, -1]

这些是按字典顺序排列的非零系数。要以匹配顺序查看单项式，请使用

>>> p.monoms()
[(3, 0, 0), (2, 1, 0), (2, 0, 1), (1, 2, 0), (1, 1, 1), (1, 0, 2), (0, 3, 0), (0, 2, 1), (0, 1, 2), (0, 0, 3)]

要得到特定单项式的系数，请使用

>>> p.coeff_monomial(x**2*y)
6

- user3717023

---

不必依赖于隐式变量顺序，可以按照任何你想要的顺序将变量传递给poly，例如 poly((x+2*y-z)**3, x, y, z)。 - asmeurer

Answer 2

一个多项式的单项式按生成元出现的顺序列出（并且该顺序由用户控制）：

>>> from sympy import Poly
>>> from sympy.abc import x, y, z
>>> Poly(x + 3*y**2, x, y).monoms()
[(1, 0), (0, 2)]
>>> Poly(x + 3*y**2, y, x).monoms()
[(2, 0), (0, 1)]

查询系数时，可以使用单项式元组或表达式：

>>> Poly(x + 3*y**2, x, y).coeff_monomial(y**2)
3
>>> Poly(x + 3*y**2, x, y).coeff_monomial((0, 2))
3

通过将多项式转换为表达式，然后使用as_coefficients_dict方法，可以获得不同单项式的所有系数的字典形式：

>>> p = Poly((x+2*y-z)**3)
>>> p.as_expr().as_coefficients_dict()
defaultdict(<class 'int'>, {
x**3: 1, z**3: -1, y**3: 8, y**2*z: -12, x**2*z: -3,
x*z**2: 3, x**2*y: 6, y*z**2: 6, x*y**2: 12, x*y*z: -12})

或者，如果您喜欢单项式元组形式，可以使用：

>>> [(m,p.coeff_monomial(m)) for m in p.monoms()]
[((3, 0, 0), 1), ((2, 1, 0), 6), ((2, 0, 1), -3), ((1, 2, 0), 12), ((1, 1, 1),
-12), ((1, 0, 2), 3), ((0, 3, 0), 8), ((0, 2, 1), -12), ((0, 1, 2), 6),
((0, 0, 3), -1)]

可以使用defaultdict将其转换为一个函数，对于不存在的单项式会返回0：

>>> defaultdict(int, _)
defaultdict(<class 'int'>, {(3, 0, 0): 1, (2, 1, 0): 6, (2, 0, 1): -3,
(1, 2, 0): 12, (1, 1, 1): -12, (1, 0, 2): 3, (0, 3, 0): 8, (0, 2, 1): -12,
(0, 1, 2): 6, (0, 0, 3): -1})

- smichr