pycharm简单的线性回归模型实例

使用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个小时