Topic
線性回歸執行個體
#導入相關子產品
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn import linear_model
#導入資料源
df = pd.read_csv(r"C:\Users\Zstrong\Downloads\honeyproduction (1).csv")
#計算totalprod每年的平均值
prod_per_year = df.groupby(['year'])['totalprod'].mean()
#Data X
X = prod_per_year.index
X = X.values.reshape(-1,1)
#Data Y
y = prod_per_year
#畫資料Sheet散點圖
plt.scatter(X,y)
#sklearn 線性模型
regr = linear_model.LinearRegression()
regr.fit(X,y)
y_predict = regr.predict(X)
plt.plot(X,y_predict)
X_future = np.array(range(2013,2050))
X_future = X_future.reshape(-1,1)
future_predict = regr.predict(X_future)
plt.plot(X_future,future_predict)
plt.show()
![](https://img.laitimes.com/img/_0nNw4CM6IyYiwiM6ICdiwiIwczX0xiRGZkRGZ0Xy9GbvNGL2EzXlpXazxSP9cnTwUFRNVTT61EM4wmYwhGWhxGZzwEMW1mY1RzRapnTtxkb5ckYplTeMZTTINGMShUYfRHelRHLwEzX39GZhh2css2RkBnVHFmb1clWvB3MaVnRtp1XlBXe0xyayFWbyVGdhd3LcV2Zh1Wa9M3clN2byBXLzN3btg3Pn5GcuYDO5IDMzEjM3ETMxgTMwIzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)