在sympy

Question

我有一个多元多项式（在一般情况下有许多变量），其系数列出了我需要读取的一些数据，但它看起来并不像是一个很好的方法来做到这一点。

collect函数似乎是正确的想法，但是当你将它与几个变量一起使用时，它实际上并没有给你单个的单项式，而是依赖于你的顺序的奇异的单项式分组列出了变量。

有谁知道这样做的方法？

Answer 1

documentation of polynomial module列出了很多处理系数的方法。例如：

>>> 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

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

通过将Poly转换为表达式然后使用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})