python 最小二乘法 线性实现

.

// An highlighted block
def lesquare(data):
    """
    函数说明：
        使用最小二乘法估计参数关系
    parameters:
        data:
            DataFrame,最后1列是估计值
    return:
    		a:估计的系数
    """
    columns=list(data.columns)
    featurs=columns[:-1]
    label=columns[-1]
    x=data[featurs].values
    y=data[label].values
    m=len(data)
    x1=np.ones((m, 1))
    x= np.hstack([x1, x])
    #求系数
    a=np.linalg.inv(x.T.dot(x)).dot(x.T).dot(y)
    return a;