/
ssmih
/
Python
Обзор
Документация
Войти
/
ssmih
/
Python
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
linear_algebra/matrix_inversion.py
36 строк
940 B
Naitik Dwivedi
Add matrix inversion algorithm using NumPy (#12657)
14 апр 2025, 21:57
Не верифицирован
14 апр 2025, 21:57
4282063
Код
Авторство
О чём код?
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. >>> invert_matrix([[4.0, 7.0], [2.0, 6.0]]) [[0.6000000000000001, -0.7000000000000001], [-0.2, 0.4]] >>> invert_matrix([[1.0, 2.0], [0.0, 0.0]]) Traceback (most recent call last): ... ValueError: Matrix is not invertible """ np_matrix = np.array(matrix) try: inv_matrix = np.linalg.inv(np_matrix) except np.linalg.LinAlgError: raise ValueError("Matrix is not invertible") return inv_matrix.tolist() if __name__ == "__main__": mat = [[4.0, 7.0], [2.0, 6.0]] print("Original Matrix:") print(mat) print("Inverted Matrix:") print(invert_matrix(mat))