python解一元二次方程的代码

在Python中，你可以使用`cmath`模块或者第三方库如`numpy`来解一元二次方程。这里以`cmath`为例，假设有一个标准形式的一元二次方程ax^2 + bx + c = 0，可以使用求根公式（也称为韦达定理）来找到它的两个解。以下是简单的代码：

import cmath

def solve_quadratic_equation(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, root2
    elif discriminant == 0:
        # 有一个实数解
        root = -b / (2*a)
        return root, root
    else:   # 两个复数解
        real_part = -b / (2*a)
        imaginary_part = cmath.sqrt(-discriminant) / (2*a)
        root1 = complex(real_part, imaginary_part)
        root2 = complex(real_part, -imaginary_part)
        return root1, root2

# 示例
a, b, c = 1, -3, 2
solutions = solve_quadratic_equation(a, b, c)
print(f"一元二次方程 {a}x^2 + {b}x + {c} 的解为: {solutions}")