python一元二次方程求复数根

在Python中，如果你需要求解一元二次方程（如 ax^2 + bx + c = 0），并可能得到复数根，你可以利用cmath模块，它提供了复数运算的功能。这里是一个简单的示例：

import cmath

def quadratic_equation(a, b, c):
    # 计算判别式
    discriminant = (b**2) - (4*a*c)

    # 实部和虚部
    real_root = (-b) / (2*a)
    imaginary_root = cmath.sqrt(-discriminant) / (2*a)

    if discriminant > 0:
        roots = (real_root, real_root)  # 两个实根
    elif discriminant == 0:
        roots = (real_root, )  # 单一实根
    else:  # 两个共轭复根
        roots = (real_root, imaginary_root)

    return roots

# 示例
a = 1
b = -3
c = 2
result = quadratic_equation(a, b, c)
print("一元二次方程 {}x^2 + {}x + {} 的根为:".format(a, b, c), result)

在这个函数里，如果判别式大于零，有两个不同的实数根；等于零时有一个实数根；小于零时，会返回一对共轭复数。