高斯列主元消去法的python实现

高斯列主元消去法的原理

首先要了解高斯消去法的原理和代码，高斯消去法的原理和python实现，高斯列主元消去法是高斯消去法的改进。

高斯列主元消去法的python代码

import numpy as np

def swap(a, b, k, n):           # 找到主元并交换，这仅是一个仅用来交换的函数
    ans = 0
    for i in range(k, n):
        if ans < np.fabs(a[i][k]):    #fabs是绝对值，将a中绝对值最大的找出来
            ans = a[i][k]
            maxn = i
    a[[k, maxn], :] = a[[maxn, k], :]     #交换
    b[k], b[maxn] = b[maxn], b[k]

#主算法
def gaussin(a, b):
    cout = 0                         #定义计算次数
    m, n = a.shape                  #矩阵a的行数和列数
    if ( m < n ):
        print("There is a 解空间。")#保证方程个数大于未知数个数
    else:
        l = np.zeros((n,n))
        for i in range(n):
            # 限制条件
            if (a[i][i] == 0):
                print("no answer")
        # j表示列
        for k in range(n - 1):          # k表示第一层循环，(0，n-1)行
            swap(a, b, k, n)            #在每次计算前，找到最大主元，进行换行
            for i in range(k + 1, n):   # i表示第二层循环,(k+1,n)行,计算该行消元的系数
                l[i][k] = a[i][k] / a[k][k]     #计算
                cout += 1
                for j in range(m):      # j表示列，对每一列进行运算
                    a[i][j] = a[i][j] - l[i][k] * a[k][j]
                    cout += 1
                b[i] = b[i] - l[i][k] * b[k]
        # 回代求出方程解
        x = np.zeros(n)
        x[n - 1] = b[n - 1] / a[n - 1][n - 1] #先算最后一位的x解

        for i in range(n - 2, -1, -1):      #依次回代倒着算每一个解
            for j in range(i + 1, n):
                b[i] -= a[i][j] * x[j]
                               #自增自减
            x[i] = b[i] / a[i][i]
        for i in range(n):
            print("x" + str(i + 1) + " = ", x[i])
        print("x" " = ", x)
        print("计算次数","=",cout)


if __name__ == '__main__':      #当模块被直接运行时，以下代码块将被运行，当模块是被导入时，代码块不被运行。
    a = np.array([[0.5, 1.1, 3.1], [2.0, 4.5, 0.36], [5.0, 0.96, 6.5]])
    b = np.array([-6.0, 0.020, 0.96])
    gaussin(a, b)