import numpy as np
import statsmodels.api as sm
import pylab as plt
a = np.loadtxt('F:\python数学建模与算法\源程序\《Python数学建模算法与应用》程序和数据//10第10章 回归分析/data10_2.txt')
plt.rc('font', family='SimHei')
plt.rc('axes', unicode_minus=False)
plt.plot(a[0], a[2], '*', label='$x_1$')
plt.plot(a[1], a[2], 'o', label='$x_2$')
plt.legend(loc='upper left')
X = sm.add_constant(a[:2].T)
re = sm.OLS(a[2], X).fit()
print(re.summary())
yh = re.predict(np.array([[1,9,10],[1,10,9]]))
print('残差的方差:', re.mse_resid)
print('预测值：', yh); plt.show()
