NumPy linalg.solve() 函数

 NumPy 函数

NumPy linalg.solve() 函数用于求解线性矩阵方程或线性标量方程组。使用该函数的语法如下:

语法

numpy.linalg.solve(a, b) 

参数

返回值

将解返回给系统a x = b。返回的形状与 b 相同。

异常

如果 a 是单数或不是正方形，则引发 LinAlgError 异常。

示例:求解线性方程组

考虑下面提到的线性方程组:

可以用矩阵形式表示如下。

上面的解方程为:x=1，y=5，z=7。这可以使用 numpy.linalg.solve() 函数来解决，如下例所示。

import numpy as np
A = np.array([[1, 2, 3], 
              [2, 3, 1],
              [3, 1, 2]])
B = np.array([32, 24, 22])

sol = np.linalg.solve(A, B)
print(sol) 

上述代码的输出将是:

[ 1.  5.  7.]