动手学数据分析系列---数据清洗及特征处理
【回顾&引言】前面一章的内容大家可以感觉到我们主要是对基础知识做一个梳理,让大家了解数据分析的一些操作,主要做了数据的各个角度的观察。那么在这里,我们主要是做数据分析的流程性学习,主要是包括了数据清洗以及数据的特征处理,数据重构以及数据可视化。这些内容是为数据分析最后的建模和模型评价做一个铺垫。
开始之前,导入numpy、pandas包和数据
#加载所需的库 import numpy as np import pandas as pd #加载数据train.csv df = pd.read_csv('train.csv') df.head()
2 第二章:数据清洗及特征处理
我们拿到的数据通常是不干净的,所谓的不干净,就是数据中有缺失值,有一些异常点等,需要经过一定的处理才能继续做后面的分析或建模,所以拿到数据的第一步是进行数据清洗,本章我们将学习缺失值、重复值、字符串和数据转换等操作,将数据清洗成可以分析或建模的样子。
2.1.1 任务一:缺失值观察
(1) 请查看每个特征缺失值个数
df.info()
df.isnull().sum()
(2) 请查看Age, Cabin, Embarked列的数据
df[['Age','Cabin','Embarked']].head(5)
2.1.2 任务二:对缺失值进行处理
(1)处理缺失值一般有几种思路
(2) 请尝试对Age列的数据的缺失值进行处理
(3) 请尝试使用不同的方法直接对整张表的缺失值进行处理
#处理缺失值的一般思路: #提醒:可使用的函数有--->dropna函数与fillna函数 df[df['Age']==None]=0 df.head() df[df['Age'].isnull()] = 0 df.head() df[df['Age'] == np.nan] = 0 df.head() df.dropna().head(5) df.fillna(0).head()
2.2.1 任务一:请查看数据中的重复值
df[df.duplicated()]
2.2.2 任务二:对重复值进行处理
df = df.drop_duplicates()
df.head()
2.2.3 任务三:将前面清洗的数据保存为csv格式
df.to_csv('test_clear_data.csv')
2.3.1 任务一:对年龄进行分箱(离散化)处理
(1) 分箱操作是什么?
分箱操作就是将连续数据转换为分类对应物的过程。⽐如将连续的身⾼数据划分为:矮中⾼。分箱操作分为等距分箱和等频分箱。分箱操作也叫⾯元划分或者离散化。
(2) 将连续变量Age平均分箱成5个年龄段,并分别用类别变量12345表示
df['AgeBand'] = pd.cut(df['Age'], 5,labels = [1,2,3,4,5]) df.head()
(3) 将连续变量Age划分为[0,5) [5,15) [15,30) [30,50) [50,80)五个年龄段,并分别用类别变量12345表示
df['AgeBand'] = pd.cut(df['Age'],[0,5,15,30,50,80],labels = [1,2,3,4,5]) df.head()
(4) 将连续变量Age按10% 30% 50% 70% 90%五个年龄段,并用分类变量12345表示
df['AgeBand'] = pd.qcut(df['Age'],[0,0.1,0.3,0.5,0.7,0.9],labels = [1,2,3,4,5]) df.head()
【参考】https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.qcut.html
2.3.2 任务二:对文本变量进行转换
(1) 查看文本变量名及种类
df['Sex'].value_counts()
(2) 将文本变量Sex, Cabin ,Embarked用数值变量12345表示
df['Sex_num'] = df['Sex'].replace(['male','female'],[1, 2]) df.head()
(3) 将文本变量Sex, Cabin, Embarked用one-hot编码表示
#将类别文本转换为one-hot编码 #方法: OneHotEncoder for item in ["Age", "Embarked"]: x = pd.get_dummies(df[item], prefix=item) df = pd.concat([df, x], axis=1) #df[feat] = pd.get_dummies(df[feat], prefix=feat) df.head()
2.3.3 任务三:从纯文本Name特征里提取出Titles的特征(所谓的Titles就是Mr,Miss,Mrs等)
df['Title'] = df.Name.str.extract('([A-Za-z]+)\.', expand=False) df.head()
#保存最终你完成的已经清理好的数据 df.to_csv('test_final_data.csv')