在Python中使用Gauss-Jordan消元法求矩阵的逆矩阵

import numpy as np

A = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 10]])

rows, cols = A.shape

augmented_matrix = np.hstack((A, np.eye(rows)))

for i in range(rows):
    if augmented_matrix[i, i] == 0:
        for j in range(i + 1, rows):
            if augmented_matrix[j, i] != 0:
                augmented_matrix[[i, j]] = augmented_matrix[[j, i]]
                break
    pivot = augmented_matrix[i, i]
    augmented_matrix[i] /= pivot
    for j in range(rows):
        if j != i:
            augmented_matrix[j] -= augmented_matrix[j, i] * augmented_matrix[i]

inverse_matrix = augmented_matrix[:, cols:]