Python求平方根(附带源码)
对于求一个浮点数的平方根,可以使用库函数,如 math.sqrt()。但在本节,为了演示浮点数的一些用法,我们用下面的迭代法来求某个数的平方根。
其基本原理如下,对于一个浮点数 a,其平方根一定在 1 和 a 之间:
假定a>1,我们的方法是取 1 和 a 的中间值 a1=(a+1)/2,如果 a1 的平方大于 a,那么平方根一定在 1 和 a1 之间;否则在 a1 和 a 之间。依次迭代直到误差足够小。
下面是实现代码:
import sys # 用来获得用户输入参数 def cal_sqrt_with_newton(start, end, target, stop): # 计算平方根 while True: # 一直循环 mid = (start + end) / 2.0 if abs((mid * mid) - target) < stop: # 如果精度达到要求了 return mid # 返回中间值 else: if (mid * mid) > target: # 否则继续二分法 end = mid else: start = mid def py_sqrt(v): if v > 1.0: # 如果值大于1,那么在1和v之间 return cal_sqrt_with_newton(1.0, v, v, 1.0e-6) else: # 如果值小于1,那么在v和1之间 return cal_sqrt_with_newton(v, 1.0, v, 1.0e-6) def test(v): # 测试计算结果 ret = py_sqrt(v) print("sqrt(%s) = %s" % (v, ret)) # 显示计算结果 if __name__=='__main__': input = float(sys.argv[1]) test(input)
运行后的结果如下:
$ python sqrt2.py 2.0 # 计算2的平方根 sqrt(2.0) = 1.41421365738 # 计算结果 $ python sqrt2.py 0.5 # 计算0.5的平方根 sqrt(0.5) = 0.707106590271 # 计算结果
需要注意的是,由于使用了递归调用,如果希望得到高精度的结果,就有可能出现调用深度超出。
本文链接:http://so.lmcjl.com/news/249/