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

 

請定義一個函數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)
# 測試成功