基于Python实现Jacobi迭代法、Gauss-Seidel迭代法和SOR法方程组（三）

标题基于Python实现SOR法方程组计算（三）

一、首先根据习题的方程完成下面的代码

A = np.array([
[5,2,1],
[-1,4,2],
[2,-3,10]
])
b = np.array([-12,20,3])
w = 0.9
k=20
SORfunction(A,b,w,k)

二、整个的代码如下：

import numpy as np
def SORfunction(A,b,w,k):
    n = A.shape[1]#系数矩阵的列数
    D = np.eye(n)
    D[np.arange(n),np.arange(n)] = A[np.arange(n),np.arange(n)]
    LU = D - A
    L = np.tril(LU)
    U = np.triu(LU)
    D_wL = D - w*L
    X = np.zeros(n)
    print("第",0,"次迭代的结果：",X)
    for i in range(k):
        D_wL_inv = np.linalg.inv(D_wL)
        X =  np.dot(np.dot(D_wL_inv,w * U + (1-w) * D),X) + w * np.dot(D_wL_inv,b)
        print("第",i+1,"次迭代的结果：",X)
    return X
A = np.array([
    [5,2,1],
    [-1,4,2],
    [2,-3,10]
])
b = np.array([-12,20,3])
w = 0.9
k=20
SORfunction(A,b,w,k)