如何判断SymPy变量是否为复数？实复根分离问题求助

嘿，我懂你遇到的困扰——SymPy解方程返回的根，哪怕是实根，很多时候也会被归类为SymPy自己的复数类型，所以用Python原生的isinstance(i, complex)去判断肯定全返回True，这是因为SymPy的数值体系和Python内置类型不完全兼容。下面给你几个靠谱的解决方法：

方法一：使用SymPy自带的is_real属性 #

这是最直接也最可靠的方式，SymPy的每个数值对象都有is_real属性，会自动判断该数是否为实数（包括虚部为零的复数形式）。示例代码如下：

import sympy as sp

# 示例：求解x^7 - 1 = 0的根
x = sp.symbols('x')
roots = sp.solve(sp.Eq(x**7 - 1, 0), x)

real_roots = []
complex_roots = []

for root in roots:
    if root.is_real:
        real_roots.append(root)
    else:
        complex_roots.append(root)

print("实根列表：", real_roots)
print("复根列表：", complex_roots)

方法二：判断虚部是否为零（处理浮点精度问题） #

如果你的根是浮点形式（比如你提到的-0.000581431210287302 - 0.2540334478167*I），可以通过获取虚部并判断其是否足够小（因为数值计算可能会有微小的误差）：

import sympy as sp

x = sp.symbols('x')
roots = sp.solve(sp.Eq(x**7 - 1, 0), x)

real_roots = []
complex_roots = []
# 设置精度阈值，可根据需求调整
epsilon = 1e-12

for root in roots:
    # 获取虚部的绝对值
    imag_part = abs(sp.im(root))
    if imag_part < epsilon:
        real_roots.append(root)
    else:
        complex_roots.append(root)

print("实根列表：", real_roots)
print("复根列表：", complex_roots)

为什么isinstance不好用？ #

SymPy中的实根可能是Integer、RealNumber等类型，而复根是ComplexNumber类型，但这些都不是Python原生的complex类。不过ComplexNumber对象如果虚部为零，它的is_real属性会返回True，所以直接用这个属性比类型判断更准确。

内容的提问来源于stack exchange，提问作者G S Srivatsa