如何在sympy中提取所有系数

Question

您可以使用coeff（）获得特定项的系数；

x, a = symbols("x, a")
expr = 3 + x + x**2 + a*x*2
expr.coeff(x)
# 2*a + 1

在这里，我想提取x，x ** 2（等等）的所有系数，例如；

# for example
expr.coefficients(x)
# want {1: 3, x: (2*a + 1), x**2: 1}

有一个方法as_coefficients_dict（），但是这似乎不符合我想要的方式。

expr.as_coefficients_dict()
# {1: 3, x: 1, x**2: 1, a*x: 2}
expr.collect(x).as_coefficients_dict()
# {1: 3, x**2: 1, x*(2*a + 1): 1}

Answer 1

最简单的方法是使用 Poly

>>> a = Poly(expr, x)
>>> a.coeffs()
[1, 2*a + 1, 3]