我正在使用polyfit
将数据放到一行。该等式的等式为y = mx + b
形式。我试图检索斜率上的误差和y截距的误差。这是我的代码:
fit, res, _, _, _ = np.polyfit(X,Y,1, full = True)
此方法返回残差。但我不想要残差。所以这是我使用的另一种方法:
slope, intercept, r_value, p_value, std_err = stats.linregress(X,Y)
我知道std_err
会在斜率上返回错误。我仍然需要得到y截距的标准偏差。我怎么能这样做?
答案 0 :(得分:5)
如果可以使用最小二乘拟合,则可以使用以下函数计算斜率,y轴截距,相关系数,斜率的标准偏差和y截距的标准偏差:
import numpy as np
def lsqfity(X, Y):
"""
Calculate a "MODEL-1" least squares fit.
The line is fit by MINIMIZING the residuals in Y only.
The equation of the line is: Y = my * X + by.
Equations are from Bevington & Robinson (1992)
Data Reduction and Error Analysis for the Physical Sciences, 2nd Ed."
pp: 104, 108-109, 199.
Data are input and output as follows:
my, by, ry, smy, sby = lsqfity(X,Y)
X = x data (vector)
Y = y data (vector)
my = slope
by = y-intercept
ry = correlation coefficient
smy = standard deviation of the slope
sby = standard deviation of the y-intercept
"""
X, Y = map(np.asanyarray, (X, Y))
# Determine the size of the vector.
n = len(X)
# Calculate the sums.
Sx = np.sum(X)
Sy = np.sum(Y)
Sx2 = np.sum(X ** 2)
Sxy = np.sum(X * Y)
Sy2 = np.sum(Y ** 2)
# Calculate re-used expressions.
num = n * Sxy - Sx * Sy
den = n * Sx2 - Sx ** 2
# Calculate my, by, ry, s2, smy and sby.
my = num / den
by = (Sx2 * Sy - Sx * Sxy) / den
ry = num / (np.sqrt(den) * np.sqrt(n * Sy2 - Sy ** 2))
diff = Y - by - my * X
s2 = np.sum(diff * diff) / (n - 2)
smy = np.sqrt(n * s2 / den)
sby = np.sqrt(Sx2 * s2 / den)
return my, by, ry, smy, sby
print lsqfity([0,2,4,6,8],[0,3,6,9,12])
输出:
(1, 0, 1.0, 0.0, 2.4494897427831779)
该功能由Filipe P. A. Fernandes撰写,最初发布于here。