numpy应用之求解方程（线性方程组、超定方程组）

解方程

• 需引如下库

import numpy as np
from numpy.linalg import lstsq # 解超定方程
from numpy.linalg import solve # 解线性方程

一、线性方程

a = np.mat([[2, 3], [1, 3]])#系数矩阵
b = np.mat([5, 3]).T    #常数项列矩阵
x = solve(a, b)        #方程组的解
print(x)

二、超定方程

a = np.mat([[2, 3], [1, 3], [1, 1]])
b = np.mat([5, 3, 2]).T
x = lstsq(a, b, rcond=None)
print(x)

三、如何获取解的每一个值

• 线性方程的解如下图所示
  
• 得到线性方程的每一个解

x0 = x[0].item(0)
x1 = x[1].item(0)

• 超定方程的解如下图所示
  
• 得到超定方程的每一个解

x0 = x[0].item(0)
x1 = x[0].item(1)