python实现正规方程

用python实现正规方程

我们需要使用numpy，这是一个很方便的数学运算库，没有的话可以打开cmd输入装一个

pip install numpy

然后我们只需要导入numpy

import numpy as np

那么求解可逆矩阵的逆我们可以用

numpy.linalg.inv(a)

其中a就是我们需要求逆的矩阵，在numpy里面可以用ndarray来实现矩阵

简单举一个例子

X=np.array([[1,1],[1,1]])
print(X)

[[1 1]
 [1 1]]

如果该矩阵是奇异矩阵的话就使用

numpy.linalg.pinv(a)

同时python中可以使用@来完成矩阵之间的点乘，这个特性是在python3.5新加的，如果使用版本没有这个功能，可以用dot函数代替。

那么接下来我们只需

 def normalEqn(X, y):
   #可逆就选下面
   theta = np.linalg.inv(X.T@X)@X.T@y #X.T@X
   #不可逆就选这个
   theta = np.linalg.pinv(X.T@X)@X.T@y #X.T@X
   return theta

顺带一提pinv在可逆的时候也可以得到inv的结果，只是会慢一点而已