Add matrix inversion algorithm using NumPy

linear_algebra/matrix_inversion.py

@@ -0,0 +1,27 @@
+import numpy as np
+
+
+def invert_matrix(matrix: list[list[float]]) -> list[list[float]]:
+    """
+    Returns the inverse of a square matrix using NumPy.
+
+    Parameters:
+    matrix (list[list[float]]): A square matrix.
+
+    Returns:
+    list[list[float]]: Inverted matrix if invertible, else raises error.
+    """
+    try:
+        np_matrix = np.array(matrix)
+        inv_matrix = np.linalg.inv(np_matrix)
+        return inv_matrix.tolist()
+    except np.linalg.LinAlgError:
+        raise ValueError("Matrix is not invertible")
+
+
+if __name__ == "__main__":
+    mat = [[4.0, 7.0], [2.0, 6.0]]
+    print("Original Matrix:")
+    print(mat)
+    print("Inverted Matrix:")
+    print(invert_matrix(mat))