代码是自己在学习网课时跟着敲的,数据源data.csv文件我会放在我的资源里,有兴趣的可以自己下载试试
from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt
#载入数据,delimite表示分隔符为逗号
data=np.genfromtxt("data.csv",delimiter=",")
x_data=data[:,0]
y_data=data[:,1]
plt.scatter(x_data,y_data)
plt.show()
print(x_data.shape)
x_data=data[:,0,np.newaxis]
y_data=data[:,1,np.newaxis]
#创建并拟合模型
model=LinearRegression()
model.fit(x_data,y_data)
#画图
plt.plot(x_data,y_data,'b.')
plt.plot(x_data,model.predict(x_data),'r')
plt.show()
最终运行结果如下:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190925091819479.png =420x?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ1MzEzNjA0,size_10,color_FFFFFF,t_70)
注:如有侵权,请联系删除!