数值分析实验 实验4-1 高斯-塞德尔迭代法 python3实现

题目

代码

import numpy as np

# 系数矩阵A的行数
m = int(input("请输入系数矩阵A的行数："))
# 系数矩阵A的列数
n = int(input("请输入系数矩阵A的列数："))

# 系数矩阵A
a = np.empty([m, n])
print("请输入系数矩阵A：")
for i in range(m):
    a[i] = np.array(list(map(float, input().split())))

# 系数矩阵B
b = np.empty([m, 1])
print("请输入系数矩阵B：")
for i in range(m):
    b[i] = np.array(list(map(float, input().split())))

# 最大迭代次数
max_N = int(input("请输入最大迭代次数："))

print("请输入初始向量：")
# 初始向量
x0 = np.empty([m, 1])
for i in range(m):
    x0[i] = np.array(list(map(float, input().split())))

# 误差限
e = float(input("请输入误差限："))

# 系数矩阵 D + L
d_l = np.tril(a)

# 系数矩阵 D + L 的逆矩阵
d_l_1 = np.linalg.inv(d_l)

# 系数矩阵 U
u = np.triu(a, 1)

# 存放上次迭代的解
last_x = x0

# 当前迭代次数
current_N = 1
while True:
    if current_N > max_N:
        print("超出迭代次数！")
        break

    xk1 = np.dot(d_l_1, b) - np.dot(np.dot(d_l_1, u), last_x)

    if abs(xk1 - last_x).max() < e:
        print("解向量为：" + str(xk1))
        break

    last_x = xk1
    current_N += 1

运行结果