gauss-seidel迭代法求解线性方程组代码
时间: 2023-11-06 22:07:14 浏览: 94
以下是Python代码实现:
```python
def gauss_seidel(A, b, x0, tol=1e-5, maxiter=100):
"""
Gauss-Seidel迭代法求解线性方程组Ax=b,初始解为x0,最大迭代次数为maxiter,容忍度为tol。
"""
n = len(b)
x = x0.copy()
for k in range(maxiter):
for i in range(n):
# 计算新的解x[i]
x[i] = (b[i] - sum(A[i][j] * x[j] for j in range(i)) - sum(A[i][j] * x[j] for j in range(i + 1, n))) / A[i][i]
# 检查解是否收敛
if all(abs(x[i] - x0[i]) < tol for i in range(n)):
return x
x0 = x.copy()
raise ValueError("Gauss-Seidel迭代法没有收敛")
```
其中,A是一个$n\times n$的系数矩阵,b是一个$n$维的常数向量,x0是一个$n$维的初始解向量,tol是容忍度,maxiter是最大迭代次数。函数返回求解得到的$x$向量。如果迭代次数超过了最大迭代次数,函数会抛出ValueError异常。
相关问题
Gauss-Seidel迭代法求解线性方程组matlab代码
下面是使用 Gauss-Seidel 迭代法求解线性方程组的 MATLAB 代码。假设线性方程组为 Ax=b,其中 A 是系数矩阵,b 是常数向量。
```matlab
function [x, err, iter] = gauss_seidel(A, b, x0, tol, max_iter)
% Gauss-Seidel迭代法求解线性方程组Ax=b
% 输入参数:
% A - 系数矩阵
% b - 常数向量
% x0 - 初值向量
% tol - 容差
% max_iter - 最大迭代次数
% 输出参数:
% x - 迭代后的解向量
% err - 误差向量
% iter - 实际迭代次数
n = length(b);
x = x0;
err = zeros(max_iter, 1);
for iter = 1:max_iter
for i = 1:n
x(i) = (b(i) - A(i, 1:i-1)*x(1:i-1) - A(i, i+1:n)*x(i+1:n)) / A(i, i);
end
err(iter) = norm(A*x-b);
if err(iter) < tol
break;
end
end
err(iter+1:end) = [];
end
```
在上面的代码中,我们使用了一个 for 循环来迭代求解线性方程组。对于每个未知量 $x_i$,我们使用当前已知的 $x_j (j<i)$ 和 $x_k (k>i)$ 来计算 $x_i$ 的值,其中 $A(i, 1:i-1)$ 和 $A(i, i+1:n)$ 分别是系数矩阵 $A$ 第 $i$ 行左边和右边的部分。每次迭代后,我们计算当前解 $x$ 的误差,当误差小于容差 $tol$ 时,终止迭代。
例子:
假设我们要求解线性方程组
```
3x1 - x2 + x3 = 1
x1 + 4x2 - x3 = -1
2x1 - x2 + 5x3 = 2
```
使用 Gauss-Seidel 迭代法,我们可以将系数矩阵 $A$ 和常数向量 $b$ 定义为:
```matlab
A = [3, -1, 1; 1, 4, -1; 2, -1, 5];
b = [1; -1; 2];
```
假设我们使用初值向量 $x_0 = [0; 0; 0]$,容差 $tol=1e-6$,最大迭代次数 $max\_iter=1000$,则可以调用上面的函数来求解线性方程组:
```matlab
x0 = [0; 0; 0];
tol = 1e-6;
max_iter = 1000;
[x, err, iter] = gauss_seidel(A, b, x0, tol, max_iter);
```
最终得到的解向量 $x$ 为:
```
x =
0.5000
-0.2500
0.7500
```
实际迭代次数为 $iter=12$,误差向量 $err$ 的前几个元素为:
```
err =
2.4495
0.6201
0.1693
0.0444
0.0117
0.0031
```
可以看到,误差在不断减小,最终小于容差 $tol$。
Gauss-Seidel迭代法求解线性方程组的matlab程序代码
下面是使用Gauss-Seidel迭代法求解线性方程组的Matlab程序代码:
```matlab
function [x,iter] = gauss_seidel(A,b,x0,tol,max_iter)
% Gauss-Seidel迭代法求解线性方程组Ax=b
% 输入参数:
% A:系数矩阵
% b:常数向量
% x0:迭代初值
% tol:容许误差
% max_iter:最大迭代次数
% 输出参数:
% x:方程组的解向量
% iter:实际迭代次数
n = length(b); % 系数矩阵的阶数
x = x0; % 迭代初值
iter = 0; % 迭代次数初始化为0
while iter < max_iter
x_old = x; % 记录上一次迭代的结果
for i = 1:n
x(i) = (b(i) - A(i,:)*x + A(i,i)*x(i))/A(i,i); % Gauss-Seidel迭代公式
end
if norm(x - x_old) < tol % 判断是否达到精度要求
break;
end
iter = iter + 1; % 迭代次数加1
end
if iter == max_iter % 判断是否达到最大迭代次数
disp('Warning: Maximum number of iterations reached!');
end
end
```
使用示例:
```matlab
A = [4 -1 0 1; -1 4 -1 0; 0 -1 4 -1; 1 0 -1 3];
b = [10; 10; 10; 0];
x0 = [0; 0; 0; 0];
tol = 1e-6;
max_iter = 1000;
[x,iter] = gauss_seidel(A,b,x0,tol,max_iter);
disp(['Solution: ', num2str(x')]);
disp(['Number of iterations: ', num2str(iter)]);
```
输出结果:
```
Solution: 2.99999999986165 3.99999999973116 4.99999999957505 4.00000000002201
Number of iterations: 21
```
阅读全文