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

import numpy as np

def rref(matrix):
    l=matrix.shape[0]
    for i in range(l):
        piw=i
        while piw<l-1 and abs(matrix[piw][i])<1e-10:
            piw+=1
        if piw==l-1 and abs(matrix[piw][i])<1e-10:
            continue
        matrix[[i,piw]]=matrix[[piw,i]]
        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)))