jupyter写出高斯消除算法Gauss_elimination(A)

我是AI语言模型，我可以帮您生成高斯消除算法的Python代码。以下是代码示例：

def gauss_elimination(A):
    n = len(A)
    for i in range(n):
        # Find the maximum element in the ith column
        max_elem = abs(A[i][i])
        max_row = i
        for j in range(i+1, n):
            if abs(A[j][i]) > max_elem:
                max_elem = abs(A[j][i])
                max_row = j
        # Swap the maximum row with the ith row
        for k in range(i, n+1):
            temp = A[max_row][k]
            A[max_row][k] = A[i][k]
            A[i][k] = temp
        # Reduce the remaining rows
        for j in range(i+1, n):
            c = -A[j][i] / A[i][i]
            for k in range(i, n+1):
                if i == k:
                    A[j][k] = 0
                else:
                    A[j][k] += c * A[i][k]
    # Back substitution
    x = [0] * n
    for i in range(n-1, -1, -1):
        x[i] = A[i][n] / A[i][i]
        for j in range(i-1, -1, -1):
            A[j][n] -= A[j][i] * x[i]
    return x

其中，输入参数A是一个 n x (n+1) 的矩阵，其中前 n 列为系数矩阵，最后一列为常数向量。函数返回一个 n 维向量，表示方程组的解。