python求平方根的代码_Python求平方根（附带源码）

对于求一个浮点数的平方根，可以使用库函数，如 math.sqrt()。但在本节，为了演示浮点数的一些用法，我们用下面的迭代法来求某个数的平方根。

其基本原理如下，对于一个浮点数 a，其平方根一定在 1 和 a 之间：

如 a=2，那么其平方根 1.414 在 1 和 2 之间。

如 a=0.8，那么其平方根 0.89443 在 0.8 和 1 之间。

假定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 # 计算结果

需要注意的是，由于使用了递归调用，如果希望得到高精度的结果，就有可能出现调用深度超出。