使用lstsq实现简单的线性回归模型
import numpy as np
from scipy.linalg import lstsq
# 飞机飞行距离花费时间
distance = np.array([365,1462,1285,1096,517,1686,932,1160])
time = np.array([1.167,2.333,2.250,2.083,2.250,2.833,1.917,2.167])
M = distance[:, np.newaxis]**[0, 1]
model, _, _, _ = lstsq(M,time)
print("截距 =", model[0])
print("距离的系数 =", model[1])
x = 454#输入飞机飞行454公里时等
y = model[1]*x+model[0]
print(y)
打印结果如图;
可知预测结果飞行454公里需要1.61个小时