数据分析模型之一元回归Python代码

导⼊第三⽅模块

import statsmodels.api as sm
sm.ols(formula, data, subset=None, drop_cols=None)
formula：以字符串的形式指定线性回归模型的公式，如'y~x'就表示简单线性回归模型
data：指定建模的数据集
subset：通过bool类型的数组对象，获取data的⼦集⽤于建模
drop_cols：指定需要从data中删除的变量

导⼊第三⽅模块

  import pandas as pd
  import statsmodels.api as sm
  income = pd.read_csv('Salary_Data.csv')
  # income
  # 利⽤收⼊数据集，构建回归模型
  fit = sm.formula.ols('Salary ~ YearsExperience', data = income).fit()  # 一定要加fit
  fit.params