高斯消元 python
时间: 2023-10-28 11:00:11 浏览: 99
python实现高斯列组元消元
高斯消元法是一种用于解线性方程组的数值计算方法。它的核心思想是通过矩阵运算将线性方程组转化为上三角形矩阵,然后通过回代求解出未知数的值。在Python中,可以使用NumPy库来实现高斯消元法。
以下是一个使用Python实现高斯消元法的示例代码:
```python
import numpy as np
def gauss(a, b):
n, m = a.shape
x = np.empty(m) # 存放解向量
# 高斯消元
for k in range(n-1):
for i in range(k+1, n):
l = a[i][k / a[k][k]
for j in range(n):
a[i][j = a[i][j - l * a[k][j]
b[i = b[i - l * b[k]
# 回代求解
x[m-1 = b[m-1 / a[m-1][m-1]
for i in range(n-2, -1, -1):
sum = 0
for j in range(i+1, n):
sum = sum + a[i][j * x[j]
x[i = (b[i - sum) / a[i][i]
print(x)
if __name__ == '__main__':
a = np.array([[2, 1, 2], [5, -1, 1], [1, -3, -4]], dtype=float)
b = np.array([5, 8, -4], dtype=float)
gauss(a, b)
```
以上代码使用了NumPy库中的`array`函数创建了系数矩阵`a`和常数向量`b`,然后调用`gauss`函数进行高斯消元求解。最终打印出解向量`x`的值。
希望以上代码对您有所帮助。如果您对高斯消元法还有其他问题,请随时提问。
阅读全文