python对逻辑回归进行显著性检验_python 数据分析基础 day19－使用statsmodels进行逻辑回归...

今天是读《python数据分析基础》的第19天，读书笔记内容为使用statsmodels进行逻辑回归。

以下代码将按数据清洗、训练模型、得出测试集的预测值这三个步骤展示

逻辑回归模型的使用。

#使用逻辑回归预测客户流失概率

import pandas as pd

import numpy as np

import statsmodels.api as sma

#导入数据

inputCsv='数据路径'

churn=pd.read_csv(inputCsv)

#数据预处理

#将列标题的空格替换为下划线，将引号和问号去除，标题字母变为小写

churn.columns=churn.columns.str.replace(' ','_').str.replace('\'','').str.strip('?').str.lower()

#将churn字段值中'.'删除,

churn.churn=churn.churn.str.strip('.')

#print(churn.head(5))

#新增一个字段，将churn字段转换为01编码字段

churn['churn01']=np.where(churn.churn=='True',1,0)

#对字段intl_plan及vmail_plan进行独热编码(新增虚拟变量)

intl_plan_dummy=pd.get_dummies(churn.intl_plan,prefix='intl_plan')