python案例 - 求解一元二次方程

Posted 2020-12-20 我若亦如风

 

请定义一个函数quadratic(a, b, c)，接收3个参数，返回一元二次方程 ax^2+bx+c=0ax2+bx+c=0 的两个解。

提示：

一元二次方程的求根公式为：

计算平方根可以调用math.sqrt()函数：

 

#!/usr/bin/env python3

import math

def quadratic(a, b, c):
    x1=(-b + math.sqrt(b**2-4*a*c))/(2*a)
    x2=(-b - math.sqrt(b**2-4*a*c))/(2*a)
    return x1,x2

print(\'quadratic(2, 3, 1) =\', quadratic(2, 3, 1))
print(\'quadratic(1, 3, -4) =\', quadratic(1, 3, -4))

if quadratic(2, 3, 1) != (-0.5, -1.0):
    print(\'测试失败1\')
elif quadratic(1, 3, -4) != (1.0, -4.0):
    print(\'测试失败2\')
else:
    print(\'测试成功\')

# quadratic(2, 3, 1) = (-0.5, -1.0)
# quadratic(1, 3, -4) = (1.0, -4.0)
# 测试成功