写一个 n 阶矩阵的逆矩阵代码

以下是一个求 n 阶矩阵的逆矩阵的代码：

import numpy as np

def inverse_matrix(matrix):
    n = len(matrix)
    identity = np.identity(n)
    augmented_matrix = np.concatenate((matrix, identity), axis=1)
    for i in range(n):
        pivot = augmented_matrix[i][i]
        for j in range(i+1, n):
            ratio = augmented_matrix[j][i] / pivot
            augmented_matrix[j] = augmented_matrix[j] - ratio * augmented_matrix[i]
    for i in range(n-1, -1, -1):
        pivot = augmented_matrix[i][i]
        for j in range(i-1, -1, -1):
            ratio = augmented_matrix[j][i] / pivot
            augmented_matrix[j] = augmented_matrix[j] - ratio * augmented_matrix[i]
        augmented_matrix[i] = augmented_matrix[i] / pivot
    inverse = augmented_matrix[:, n:]
    return inverse

注意，这只是一个简单的实现，可能不适用于所有情况。