回归分析demo

最新推荐文章于 2024-11-07 17:15:19 发布

weixin_34248118

最新推荐文章于 2024-11-07 17:15:19 发布

阅读量71

点赞数

文章标签：人工智能 python

原文链接：https://my.oschina.net/kyo4321/blog/1068537

版权

2019独角兽企业重金招聘Python工程师标准>>>

#!/usr/bin/env python
from pandas import DataFrame, Series
import pandas as pd
import numpy as np
import statsmodels.formula.api as sm
from sklearn.linear_model import LinearRegression
import scipy, scipy.stats
import matplotlib.pyplot as plt

data_str = """Region|Alcohol|Tobacco
North|6.47|4.03
Yorkshire|6.13|3.76
Northeast|6.19|3.77
East Midlands|4.89|3.34
West Midlands|5.63|3.47
East Anglia|4.52|2.92
Southeast|5.89|3.20
Southwest|4.79|2.71
Wales|5.27|3.53
Scotland|6.08|4.51
Northern Ireland|4.02|4.56"""

d = data_str.split('\n')
d = [i.split('|') for i in d]

for i in range(len(d)):
    for j in range(len(d[0])):
        try:
            d[i][j] = float(d[i][j])
        except:
            pass
        
df = DataFrame(d[1:], columns=d[0])
plt.scatter(df['Tobacco'], df['Alcohol'],
         marker='o',
         edgecolor='b',
         facecolor='none',
         alpha=0.5)
plt.xlabel('Tobacco')
plt.ylabel('Alcohol')
plt.savefig('alcohol_v_tobacco.png', fmt='png', dpi=100)


df['Eins'] = np.ones((len(df), ))
Y = df.Alcohol[:-1]
X = df[['Tobacco','Eins']][:-1]
result = sm.OLS(Y, X).fit()
result.summary()

result.params   ##'Eins'即为截距项
"""
Out[22]: 
Tobacco    1.005896
Eins       2.041223
dtype: float64
"""

转载于:https://my.oschina.net/kyo4321/blog/1068537