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

在Python中，你可以使用`cmath`库来计算一元二次方程的复数根，如果你只关心实数根，可以使用标准库`math`结合公式法来实现。下面是一个简单的函数示例：

import cmath

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

    # 实数解
    if discriminant >= 0:
        root1 = (-b + cmath.sqrt(discriminant)) / (2 * a)
        root2 = (-b - cmath.sqrt(discriminant)) / (2 * a)
        return [root1.real, root2.real]
    
    # 复数解
    else:
        real_part = -b / (2 * a)
        imaginary_part = cmath.sqrt(-discriminant) / (2 * a)
        return [real_part, complex(real_part, imaginary_part)]

# 使用示例
coefficients = [1, -3, 2]  # ax^2 + bx + c = 0 的系数
roots = quadratic_solver(*coefficients)
print(f"一元二次方程 {coefficients[0]}x^2 + {coefficients[1]}x + {coefficients[2]} 的根是 {roots}")