Sympy:在多项式中删除高阶项

Question

使用Sympy,假设我们有一个表达式f,它是符号"x"(以及可能的其他符号)的多项式.

我想知道如果有一种有效的方法可以删除大于某个整数n的f中的所有项.

作为一个特例,我有一个非常复杂的功能,但我想只保留x中的第二个订单.有效的方法是什么？

这种明显的,非常有效的方法是每m小于n,取m个导数并将x设为0,得到x ^ m的系数.我们以这种方式获得每个系数,然后重建多项式.但是采取衍生品并不是最有效的方法.

Answer 1

一个简单的方法是添加O(x**n)到表达式,如

In [23]: x + x**2 + x**4 + x**10 + O(x**3)
Out[23]:
     2    ? 3?
x + x  + O?x ?

如果您想稍后删除它,请使用该removeO方法

In [24]: (x + x**2 + x**4 + x**10 + O(x**3)).removeO()
Out[24]:
 2
x  + x

您还可以使用series表达式的系列扩展.这里的区别是如果非多项式项最终在表达式中的行为:

In [25]: x + sin(x) + O(x**3)
Out[25]:
              ? 3?
sin(x) + x + O?x ?

In [26]: (x + sin(x)).series(x, 0, 3)
Out[26]:
       ? 3?
2?x + O?x ?