ex10.2_1

点击查看代码

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')
d = {'x1': a[0], 'x2': a[1], 'y': a[2]}
re = sm.formula.ols('y~x1+x2', d).fit()
print(re.summary())
yh = re.predict({'x1': [9, 10], 'x2': [10, 9]})
print('残差的方差:', re.mse_resid)
print('预测值：', yh); plt.show()