Python:仅打印二次方程容差范围内的结果

Question

print("input a, b and c for a quadratic equation ax^2+bx+c=0")
a = float(input("a ="))
b = float(input("b ="))
c = float(input("c ="))


D = (b**2) - (4*a*c)

if D>0: 
 s1 = (-b+D**0.5)/(2*a)
 s2 = (-b-D**0.5)/(2*a)
 print("the two solutions are: {} and {}".format(s1,s2))

elif D==0:
 s3 = (-b)/(2*a)
 print("the one solution is: {}".format(s3)) 

elif D<0:
 print("no solution")

这段代码有效，但我需要将这段代码变成一个函数，如果 a 和 c 之间的差异在公差“tol”之内，它只会打印 c，不知道如何进行。

Answer 1

您可以使用 Python 的内置 abs函数或 NumPy 的 isclose功能。
您可以添加另一个 if条件为 abs(a - c) < tol 的语句或 np.isclose(a, c) , 同时指定 atol (绝对公差)或 rtol (相对容差)可选参数为所需值。
引用:https://numpy.org/doc/stable/reference/generated/numpy.isclose.html