数值分析——不动点迭代法（Python实现）

Python代码实现

Code1.py.

// An highlighted block
import numpy as np

'''不动点迭代法'''

'根据图像，大概估计根的个数及所处区间'
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 1000)
y = x * x * x + 3 * x * x - 1
yy = [0 for i in range(len(x))]
plt.plot(x, y, 'k-')
plt.plot(x, yy, "r--")
plt.xlim(-3, 1)
plt.ylim(-2, 2)
plt.show()


def DieDai(x0, h):
    eps = 0.5e-8
    k = 0
    while (np.abs(h(x0) - x0) > eps and k < 1000):
        x0 = h(x0)
        k += 1
        print("第", k, "次迭代，截断误差为：", np.abs(h(x0) - x0))
    print(eps)
    if k == 1000:
        return ('error')
    else:
        return x0


def h(x):
    return x*x*x+3*x*x-1  # x0 = -0.5


# x0 = np.pi/2
x0 = -0.5
print(DieDai(x0, h))

运行结果