困扰了三个小时的问题总算解决了。利用solve()函数计算一个方程,结果可能会出现虚根,怎样去除虚根保留实根。
话不多说,先看个例子:求解x^4-1。我只想输出实数根
import sympy
from sympy import *
x, y = symbols('x y')
g = x**4-1
RS=solve(g,x)
RS_1=0
for i in range(0, len(RS)):
if (isinstance(RS[i],sympy.core.numbers.One) or isinstance(RS[i],sympy.core.numbers.NegativeOne)):
RS_1=RS[i];break;
print(RS_1,type(RS[i]))
在这个例子中,isinstance()这个函数,就是判断两个类型是否一致。如果是则为true,如果不是则为flase.
isinstance(object, classinfo)
但是,在sympy中,不可以使用int或者float。
int正数 | sympy.core.numbers.One |
int负数 | sympy.core.numbers.NegativeOne |
float | sympy.core.numbers.Float |
正虚根 | sympy.core.numbers.ImaginaryUnit |
负虚根 | sympy.core.mul.Mul |
这样就可以结合IF语言让它输出任何类型的根。