题解 | 实现简化行阶梯形（RREF）函数

import numpy as np

def rref(matrix):
    l=matrix.shape[0]
    h=matrix.shape[1]
    for i in range(l):
        if int(matrix[i][i])==0:
            continue
        else:
            matrix[i]=matrix[i]/matrix[i][i]
            for j in range(l):
                if j!=i:
                    matrix[j]-=matrix[j][i]*matrix[i]

    return np.round(matrix,1)


if __name__ == "__main__":
    print(rref(np.array(eval(input()),float)))