python求一元二次方程的根的代码

在Python中，你可以使用math库中的公式来计算一元二次方程的根。一元二次方程的一般形式是ax² + bx + c = 0，其中a、b和c是常数，a ≠ 0。下面是使用二次公式求解的示例：

import math

def quadratic_equation(a, b, c):
    # 计算判别式
    discriminant = b**2 - 4*a*c
    
    if discriminant > 0:
        # 有两个实根
        root1 = (-b + math.sqrt(discriminant)) / (2 * a)
        root2 = (-b - math.sqrt(discriminant)) / (2 * a)
        return (root1, root2)
    elif discriminant == 0:
        # 有一个实根（重复）
        root = -b / (2 * a)
        return (root,)
    else:
        # 两个共轭复根
        real_part = -b / (2 * a)
        imaginary_part = math.sqrt(-discriminant) / (2 * a)
        return (real_part, real_part + imaginary_part*1j, real_part - imaginary_part*1j)

# 示例：解决方程x^2 + 5x + 6 = 0
a, b, c = 1, 5, 6
roots = quadratic_equation(a, b, c)
print(f"方程 {a}x^2 + {b}x + {c} 的根为: {roots}")