使用Sympy,假设我们有一个表达式f,它是符号“x”(以及可能的其他符号)的多项式.
我想知道如果有一种有效的方法可以删除大于某个整数n的f中的所有项.
作为一个特例,我有一个非常复杂的功能,但我想只保留x中的第二个订单.有效的方法是什么?
这种明显的,非常有效的方法是每m小于n,取m个导数并将x设为0,得到x ^ m的系数.我们以这种方式获得每个系数,然后重建多项式.但是采取衍生品并不是最有效的方法.
解决方法
一个简单的方法是在表达式中添加O(x ** n)
In [23]: x + x**2 + x**4 + x**10 + O(x**3) Out[23]: 2 ⎛ 3⎞ x + x + O⎝x ⎠
In [24]: (x + x**2 + x**4 + x**10 + O(x**3)).removeO() Out[24]: 2 x + x
您还可以使用系列来进行表达式的系列扩展.这里的区别是如果非多项式项最终在表达式中的行为:
In [25]: x + sin(x) + O(x**3) Out[25]: ⎛ 3⎞ sin(x) + x + O⎝x ⎠ In [26]: (x + sin(x)).series(x,3) Out[26]: ⎛ 3⎞ 2⋅x + O⎝x ⎠