代码改变世界

Jacobi迭代

2019-10-10 15:42  生活的味道  阅读(169)  评论(0编辑  收藏  举报

收敛性:

对应的矩阵J=A-D谱小于1

import numpy as np
print('start')
N=2
M=20
#A=np.array([[2,1],[1,3]])
D=np.array([[2,0],[0,3]])
F=np.array([[0,-1],[-1,0]])
X=np.array([1,1])
B=np.array([10,10])
B=np.matmul(np.linalg.inv(D),B)
for i in range(M):
    X=np.matmul(np.linalg.inv(D),np.matmul(F,X))
    X=X+B
print(X)