Python实现用初等行变换将矩阵化为最简行形式

python实现用初等行变换将矩阵化为最简行形式

代码如下：

# -*- coding: utf-8 -*-
"""
Created on Fri Mar 15 15:05:53 2019

@author: MrZjjPolarBear
"""

import numpy as np

def rsmat(arbmat):
    """ Convert an arbitrary matrix to a simplest matrix """
    arbmat = arbmat.astype(float)
    row_number, column_number = arbmat.shape
    if row_number == 1:
        if arbmat[0, 0] != 0:
            return (arbmat/arbmat[0, 0])
        else:
            return arbmat
    else:
        rc_number = min(row_number, column_number)
        anarbmat = arbmat.copy()
        r = 0
        for n in range(rc_number):
            s_row = -1
            for i in arbmat[r:row_number, n]:
                s_row += 1
                if abs(i) > 1e-10:
                    anarbmat[r, :] = arbmat[s_row+r, :]
                    for j in range(r, row_number):
                        if j < s_row+r:
                            anarbmat[j+1, :] = arbmat[j, :]
                    arbmat = anarbmat.copy()
            if abs(anarbmat[r, n]) > 1e-10:
                anarbmat[r, :] = anarbmat[r, :] / anarbmat[r, n]
                for i in range(row_number):
                    if i != r:
                        anarbmat[i, :] -= \
                        anarbmat[i, n]*anarbmat[r, :]
            arbmat = anarbmat.copy()
            if abs(arbmat[r, n]) < 1e-10:
                r = r
            else:
                r = r + 1
        for m in range(column_number):
            if abs(arbmat[-1, m]) > 1e-10:
                arbmat[-1, :] = arbmat[-1, :]/arbmat[-1, m]
                for i in range(row_number-1):
                    arbmat[i, :] -= \
                    arbmat[i, m]*arbmat[-1, :]
                break
            
        return arbmat

a = np.matrix([
        [1, 0, 2, -1, 1, 3],
        [2, 0, 3, 1, 4, 5],
        [3, 0, 4, 3, 2, 6]])

b = np.matrix([
        [0, 2, -3, 1],
        [0, 3, -4, 3],
        [0, 4, -7, -1]])

c = np.matrix([
        [1, -1, 3, -4, 3],
        [3, -3, 5, -4, 1],
        [2, -2, 3, -2, 0],
        [3, -3, 4, -2, -1]])

d = np.matrix([
        [2, 3, 1, -3, -7],
        [1, 2, 0, -2, -4],
        [3, -2, 8, 3, 0],
        [2, -3, 7, 4, 3]])
    
e = np.mat([[0, 1, 2]])
    
a1 = rsmat(a) 
b1 = rsmat(b)
c1 = rsmat(c)
d1 = rsmat(d) 
e1 = rsmat(e)