使用Python读取mysql中的数据并调用sklearn.linear_model进行线性回归预测

码农道路深似海

于 2019-08-04 15:30:12 发布

阅读量1.7k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python读取mysql 线性回归预测

本文链接：https://blog.youkuaiyun.com/qq_36857730/article/details/98469981

这篇博客介绍了如何使用Python从MySQL数据库中提取GDP和year数据，然后利用sklearn.linear_model进行线性回归分析，预测下一年的GDP值。通过调整数据格式以适应模型，最终得出预测结果20861.57682927。

环境 Python3
要求：从mysql数据库中读取两列数据进行线性回归预测，例如从经济数据表中读取GDP列和year列，进行GDP对year的线性回归分析，并根据线性回归系数预测下一年的GDP值。
在这里插入图片描述
首先从mysql中读取数据并打印

from pymysql import *
import pandas as pd
import numpy as np
from sklearn import linear_model
from sqlalchemy import create_engine
import matplotlib.pyplot as plt

def show_linear_line(X_parameters,Y_parameters):
    regr = linear_model.LinearRegression()
    regr.fit(X_parameters, Y_parameters)
    plt.scatter(X_parameters,Y_parameters,color='blue')
    plt.plot(X_parameters,regr.predict(X_parameters),color='red',linewidth=4)
    plt.xticks(())
    plt.yticks(())
    plt.show()

'''
查询数据库某一列保存为list
 '''
conn = connect(host='10.x.xx.xxx', port=3306, database='population',
               user='root',
               password='root', charset='utf8')
cs1 = conn.cursor()
 #四个参数，表名，列名1，列名2，预测年份
input1 = 'economy'
input2