python - 使用 Sympy 删除多项式的低阶项

Question

假设我们有这个功能，

f = 聚(2*x**2 + 3*x - 1,x)

如何删除 n 或更低阶的项。

例如，如果 n = 1，结果将是 2*x**2。

Answer 1

from sympy import poly
from sympy.abc import x

p = poly(x ** 5 + 2 * x ** 4 - x ** 3 - 2 * x ** 2 + x)
print(p)
n = 2
new_p = poly(sum(c * x ** i[0] for i, c in p.terms() if i[0] > n))
print(new_p)

输出:

Poly(x**5 + 2*x**4 - x**3 - 2*x**2 + x, x, domain='ZZ')
Poly(x**5 + 2*x**4 - x**3, x, domain='ZZ')