金钱是否让人感到快乐#利用人均GDP和“幸福指数”建立模型预测国人幸福指数#python#线性回归#多项式回归

最新推荐文章于 2022-07-28 10:58:38 发布

原创

最新推荐文章于 2022-07-28 10:58:38 发布 · 915 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习

本文通过分析OECD的“幸福指数”和IMF的人均GDP数据，建立线性回归和多项式回归模型，探讨两者间的关联。结果显示，多项式回归模型能更好地拟合数据，预测我国2019年人均GDP对应的幸福指数为5.9。

项目场景及问题描述：

金钱是否让人感到快乐？这是个问题！OECD(经合组织)提供“幸福指数”数据，IMF(国际货币贸易基金组织)提供人均GDP统计数据，这两者是否存在某种关联，如果有，那么是否可以通过已有的数据预测一些国家的国民“幸福指数”呢？

解决方案：

建立单变量线性回归模型和多项式线性回归模型拟合数据，然后预测数据，得出结果。

import matplotlib
import matplotlib.pyplot as plt
import numpy as np
from sklearn import linear_model
from sklearn.preprocessing import PolynomialFeatures
import csv
# 读取数据
datasets_X = []
datasets_y = []
with open('E:\Fpc_VSCode_Python\机器学习实战\第一章\OECD_IMF_20201106.csv') as f:
    f_csv = csv.reader(f)
    headers = next(f_csv)
    for row in f_csv:<