Scipy线性回归:只计算缩放/斜率参数,截距固定为0



我正在尝试使用scipy.stats. linregression以最小二乘意义计算两组数据之间的比例因子。然而,它给了我一个截距,尽管事实上输入的xi变量是一个向量,而不是n X 2矩阵。

那么,一个简单的代码如下:

from scipy import stats
from numpy import arrange,array
y = [0, 11, 19, 28, 41, 49, 62, 75, 81]
xi = arange(0,9)
scale, intercept, r_value, p_value, std_err = stats.linregress(xi,y)

运行这个,我得到的比例为10.383,但我也得到的截距为-0.86。我怎么能告诉它只适合缩放参数和截距应该保持零。

如果你想拟合一个模型,y~xi没有截距,你可能要考虑使用更多的面向统计的包,如statsmodels:

In [17]:
import statsmodels.api as sm
import numpy as np
y = [0, 11, 19, 28, 41, 49, 62, 75, 81]
xi = np.arange(0,9)
model = sm.OLS(y, xi)
results = model.fit()
In [18]:
print results.params
[ 10.23039216]

您可以使用R独立验证结果。只是现在必须显式地指定截距为0:

x <- c(0, 1, 2, 3, 4, 5, 6, 7, 8)
y <- c(0, 11, 19, 28, 41, 49, 62, 75, 81)
model1 <- lm(y~x+0)
summary(model1)
Call:
lm(formula = y ~ x + 0)
Residuals:
    Min      1Q  Median      3Q     Max 
-2.6912 -1.4608  0.0000  0.6176  3.3873 
Coefficients:
  Estimate Std. Error t value Pr(>|t|)    
x   10.230      0.129   79.29 7.14e-13 ***
---
Signif. codes:  0 ‘***’ 0.001 ‘**’ 0.01 ‘*’ 0.05 ‘.’ 0.1 ‘ ’ 1
Residual standard error: 1.843 on 8 degrees of freedom
Multiple R-squared:  0.9987,    Adjusted R-squared:  0.9986 
F-statistic:  6286 on 1 and 8 DF,  p-value: 7.14e-13

下面的计算很简单:

In [29]:
import scipy.optimize as so
so.fmin(lambda b, x, y: ((b*x-y)**2).sum(), x0=0.1, args=(xi, y))
Optimization terminated successfully.
         Current function value: 27.171569
         Iterations: 27
         Function evaluations: 54
Out[29]:
array([ 10.23039063])

最新更新