牛顿迭代法解方程

Preface

前两天帮高中室友解一个方程,他课设中需要解的一个高次方程,使用的是牛顿迭代法,做下记录


牛顿法的具体细节本文就赘述了,可以参考文末的链接。

1.png


一句话总结下,就是通过选择点,做切线,不断迭代下去,直到收敛。

代码 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# -*- coding= utf-8 -*-

def f(x):
# f的方程
return 6900 * x ** 3 - 1767780 * x ** 2 + 250788032 * x - 120441587900

def f_first_order(x):
# f的一阶导数
return 6900 * 3 * x ** 2 - 1767780 * 2 * x + 250788032

def get_root(x0, max_iter=50, tol = 1e-8):
# 将初始值浮点化
p0 = x0 * 1.0
for i in range(max_iter):
# f的一阶导数不能为0,最普遍的说法是不能非正定
p = p0 - f(p0)/ f_first_order(p0)

# 如果小于精度值则退出迭代
if abs(p - p0) < tol:
return u'经过%s次的迭代,我们估计的参数值是%s' % (i, p)
p0 = p

print(u'已达到最大迭代次数, 但是仍然无法收敛')


# for x in range(-10000, 10000, 100):
# print(get_root(x))

参考

如何通俗易懂地讲解牛顿迭代法求开方?
python如何实现迭代法解方程?
十行代码实现牛顿方法

关注我的微信公众号[李一二],即时看更多的文章