编辑代码

# 输入一个3x3矩阵
A = [[1, 2, 3], [0, 1, 4], [5, 6, 0]]

# 创建一个3x3的单位矩阵
B = [[1, 0, 0], [0, 1, 0], [0, 0, 1]]

# 将矩阵A和单位矩阵B合并为增广矩阵AB
AB = []
for i in range(3):
    AB.append(A[i] + B[i])

# 进行高斯消元
for i in range(3):
    # 首先将主元素归一
    p = AB[i][i]
    for j in range(i, 6):
        AB[i][j] /= p
    for j in range(3):
        if j != i:
            # 将第j行的第i列元素变成0
            q = AB[j][i]
            for k in range(i, 6):
                AB[j][k] -= q*AB[i][k]

# 得到逆矩阵
A_inv = [[AB[i][j] for j in range(3, 6)] for i in range(3)]

print(A_inv)