python - python中的行梯队

Question

此代码适用于 3*3 矩阵。我需要它适用于 Ant m*n 矩阵。

import numpy as np
b=np.arange(1,10).reshape(3,3)

这检查矩阵的形状，但它只适用于 3*3 矩阵

if b.shape[0]==b.shape[1]:
    for i in range(b.shape[0]): 
        for j in range(b.shape[1]):
            if i==1:
                b[i]=b[i-i][j]*b[i]-[b[i][j]*b[i-i]]
                i=i+1
                b[i]=b[i-i][j]*b[i]-[b[i][j]*b[i-i]]
                j=j+1
                b[i]=b[i]-[b[i-1]*(b[i][j]/b[i-1][j])]
print(b)

Answer 1

实际上在 python 中有一个名为 sympy 的内置库。函数 Matrix().rref() 可用于获取矩阵的简化行阶梯形式。此函数的返回值包括两件事:1) 给定矩阵的简化行阶梯形式和 2) 矩阵中包含主元的行的索引(请注意，行是从 0 开始索引的)。

下面是一个如何使用这个函数的例子:

import sympy
sympy.Matrix([[1,2,3],[2,3,4]]).rref()

(Matrix([
[1, 0, -1],
[0, 1,  2]]), (0, 1))

您可以找到 Matrix().rref() 的实现 here .