python数据预处理的代码_【Python基础系列】常见的数据预处理方法（附代码）

↑↑↑点击上方蓝字，回复资料，嘿嘿，10个G的惊喜

本文简单介绍python中一些常见的数据预处理，包括数据加载、缺失值处理、异常值处理、描述性变量转换为数值型、训练集测试集划分、数据规范化。

1、 加载数据 1.1 数据读取

数据格式有很多，介绍常见的csv,txt,excel以及数据库mysql中的文件读取

import pandas as pddata = pd.read_csv(r'../filename.csv') #读取csv文件data = pd.read_table(r'../filename.txt') #读取txt文件data = pd.read_excel(r'../filename.xlsx') #读取excel文件# 获取数据库中的数据import pymysqlconn = pymysql.connect(host='localhost',user='root',passwd='12345',db='mydb') #连接数据库,注意修改成要连的数据库信息cur = conn.cursor() #创建游标cur.execute("select * from train_data limit 100") #train_data是要读取的数据名data = cur.fetchall() #获取数据cols = cur.description #获取列名conn.commit() #执行cur.close() #关闭游标conn.close() #关闭数据库连接col = []for i in cols:col.append(i[0])data = list(map(list,data))data = pd.DataFrame(data,columns=col)

1.2 CSV文件合并

实际数据可能分布在一个个的小的csv或者txt文档，而建模分析时可能需要读取所有数据，这时呢，需要将一个个小的文档合并到一个文件中

#合并多个csv文件成一个文件import glob#合并def hebing():csv_list = glob.glob('*.csv') #查看同文件夹下的csv文件数print(u'共发现%s个CSV文件'% len(csv_list))print(u'正在处理............')for i in csv_list: #循环读取同文件夹下的csv文件fr = open(i,'rb').read()with open('result.csv','ab') as f: #将结果保存为result.csvf.write(fr)print(u'合并完毕！')#去重def quchong(file):df = pd.read_csv(file,header=0)datalist = df.drop_duplicates()datalist.to_csv(file)if __name__ == '__main__':hebing()quchong("result.csv.csv")

1.3 CSV文件拆分

对于一些数据量比较大的文件，想直接读取或者打开比较困难，介绍一个可以拆分数据的方法吧，方便查看数据样式以及读取部分数据

##csv比较大，打不开，将其切分成一个个小文件，看数据形式f = open('NEW_Data.csv','r') #打开大文件i = 0 #设置计数器#这里1234567表示文件行数，如果不知道行数可用每行长度等其他条件来判断while i<1234567 :with open('newfile'+str(i),'w') as f1:for j in range(0,10000) : #这里设置每个子文件的大小if i < 1234567: #这里判断是否已结束，否则最后可能报错f1.writelines(f.readline())i = i+1else:break

1.4 数据查看

在进行数据分析前呢，可以查看一下数据的总体情况，从宏观上了解数据

data.head() #显示前五行数据data.tail() #显示末尾五行数据data.info() #查看各字段的信息data.shape #查看数据集有几行几列,data.shape[0]是行数,data.shape[1]是列数data.describe() #查看数据的大体情况，均值，最值，分位数值...data.columns.tolist() #得到列名的list

2、缺失值

现实获取的数据经常存在缺失，不完整的情况(能有数据就不错了，还想完整！！！)，为了更好的分析，一般会对这些缺失数据进行识别和处理

2.1 缺失值查看print(data.isnull().sum()) #统计每列有几个缺失值missing_col = data.columns[data.isnull().any()].tolist() #找出存在缺失值的列import numpy as np#统计每个变量的缺失值占比def CountNA(data):cols = data.columns.tolist() #cols为data的所有列名n_df = data.shape[0] #n_df为数据的行数for col in cols:missing = np.count_nonzero(data[col].isnull().values) #col列中存在的缺失值个数mis_perc = float(missing) / n_df * 100print("{col}的缺失比例是{miss}%".format(col=col,miss=mis_perc))

2.2 缺失值处理

面对缺失值，一般有三种处理方法：不处理、删除以及填充

2.2.1 不处理

有的算法(贝叶斯、xgboost、神经网络等)对缺失值不敏感，或者有些字段对结果分析作用不大，此时就没必要费时费力去处理缺失值啦 =。=

2.2.2 删除

在数据量比较大时候或者一条记录中多个字段缺失，不方便填补的时候可以选择删除缺失值

data.dropna(axis=0,how="any",inplace=True) #axis=0代表'行','any'代表任何空值行,若是'all'则代表所有值都为空时，才删除该行data.dropna(axis=0,inplace=True) #删除带有空值的行data.dropna(axis=1,inplace=True) #删除带有空值的列

2.2.3 填充

数据量较少时候，以最可能的值来插补缺失值比删除全部不完全样本所产生的信息丢失要少

2.2.3.1 固定值填充

data = data.fillna(0) #缺失值全部用0插补data['col_name'] = data['col_name'].fillna('UNKNOWN') #某列缺失值用固定值插补

2.2.3.2 出现最频繁值填充

即众数插补，离散/连续数据都行，适用于名义变量，如性别

freq_port = data.col_name.dropna().mode()[0] # mode返回出现最多的数据,col_name为列名data['col_name'] = data['col_name'].fillna(freq_port) #采用出现最频繁的值插补

2.2.3.3 中位数/均值插补

data['col_name'].fillna(data['col_name'].dropna().median(),inplace=True) #中位数插补，适用于偏态分布或者有离群点的分布data['col_name'].fillna(data['col_name'].dropna().mean(),inplace=True) #均值插补，适用于正态分布

2.2.3.4 用前后数据填充

data['col_name'] = data['col_name'].fillna(method='pad') #用前一个数据填充data['col_name'] = data['col_name'].fillna(method='bfill') #用后一个数据填充

2.2.3.5 拉格朗日插值法

一般针对有序的数据，如带有时间列的数据集,且缺失值为连续型数值小批量数据

from scipy.interpolate import lagrange#自定义列向量插值函数,s为列向量,n为被插值的位置,k为取前后的数据个数，默认5def ployinterp_columns(s, n, k=5):y = s[list(range(n-k,n)) + list(range(n+1,n+1+k))] #取数y = y[y.notnull()] #剔除空值return lagrange(y.index, list(y))(n) #插值并返回插值结果#逐个元素判断是否需要插值for i in data.columns:for j in range(len(data)):if (data[i].isnull())[j]: #如果为空即插值data[i][j] = ployinterp_columns(data[i],j)

2.2.3.6 其它插补方法最近邻插补、回归方法、牛顿插值法、随机森林填充等。

3、异常值

异常值是指样本中的个别值，其数值明显偏离它所属样本的其余观测值。异常值有时是记录错误或者其它情况导致的错误数据，有时是代表少数情况的正常值

3.1 异常值识别 3.1.1 描述性统计法

#与业务或者基本认知不符的数据,如年龄为负

neg_list = ['col_name_1','col_name_2','col_name_3']for item in neg_list:neg_item = data[item] < 0print(item + '小于0的有' + str(neg_item.sum())+'个')#删除小于0的记录for item in neg_list:data = data[(data[item]>=0)]

3.1.2 三西格玛法

当数据服从正态分布时，99.7%的数值应该位于距离均值3个标准差之内的距离，P(|xμ|>3σ)≤0.003

#当数值超出这个距离，可以认为它是异常值for item in neg_list:data[item + '_zscore'] = (data[item] - data[item].mean()) / data[item].std()z_abnormal = abs(data[item + '_zscore']) > 3print(item + '中有' + str(z_abnormal.sum())+'个异常值')

3.1.3 箱型图

#IQR(差值) = U(上四分位数) - L(下四分位数)#上界 = U + 1.5IQR#下界 = L-1.5IQRfor item in neg_list:IQR = data[item].quantile(0.75) - data[item].quantile(0.25)q_abnormal_L = data[item] < data[item].quantile(0.25) - 1.5*IQRq_abnormal_U = data[item] > data[item].quantile(0.75) + 1.5*IQRprint(item + '中有' + str(q_abnormal_L.sum() + q_abnormal_U.sum())+'个异常值')

3.1.4 其它

基于聚类方法检测、基于密度的离群点检测、基于近邻度的离群点检测等。

3.2 异常值处理

对于异常值，可以删除，可以不处理，也可以视作缺失值进行处理。

4、描述性变量转换为数值型

大部分机器学习算法要求输入的数据必须是数字，不能是字符串，这就要求将数据中的描述性变量(如性别)转换为数值型数据

#寻找描述变量，并将其存储到cat_vars这个list中去cat_vars = []print('\n描述变量有:')cols = data.columns.tolist()for col in cols:if data[col].dtype == 'object':print(col)cat_vars.append(col)···##若变量是有序的##print('\n开始转换描述变量...')from sklearn import preprocessingle = preprocessing.LabelEncoder()#将描述变量自动转换为数值型变量，并将转换后的数据附加到原始数据上for col in cat_vars:tran = le.fit_transform(data[col].tolist())tran_df = pd.DataFrame(tran,columns=['num_'+col])print('{col}经过转化为{num_col}'.format(col=col,num_col='num_'+col))data = pd.concat([data, tran_df], axis=1)del data[col] #删除原来的列##若变量是无序变量###值得注意的是one-hot可能引发维度爆炸for col in cat_vars:onehot_tran = pd.get_dummies(data.col)data = data.join(onehot_tran) #将one-hot后的数据添加到data中del data[col] #删除原来的列

5、训练测试集划分

实际在建模前大多需要对数据进行训练集和测试集划分，此处介绍两种划分方式

法一、直接调用train_test_split函数from sklearn.model_selection import train_test_splitX = data.drop('目标列',1) #X是特征列y = data['目标列'] #y是目标列X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.3)

法二：随机抽样

#随机选数据作为测试集test_data = data.sample(frac=0.3,replace=False,random_state=123,axis=0)#frac是抽取30%的数据，replace是否为有放回抽样，取replace=True时为有放回抽样，axis=0是抽取行、为1时抽取列#在data中除去test_data，剩余数据为训练集train_data = (data.append(test_data)).drop_duplicates(keep=False)X_train = train_data.drop('目标列',1)X_test = test_data.drop('目标列',1)y_train = train_data['目标列']y_test = test_data['目标列']

6、数据规范化

数据的标准化(normalization)是将数据按比例缩放，使之落入一个小的特定区间。在某些比较和评价的指标处理中经常会用到，去除数据的单位限制，将其转化为无量纲的纯数值，便于不同单位或量级的指标能够进行比较和加权。一些需要数据规范化的算法：LR、SVM、KNN、KMeans、GBDT、AdaBoost、神经网络等

6.1 最小最大规范化

对原始数据进行线性变换，变换到[0,1]区间。计算公式为：x* = (x-x.min)/(x.max-x.min)

from sklearn.preprocessing import MinMaxScalerx_scaler = MinMaxScaler()y_scaler = MinMaxScaler()#特征归一化x_train_sca = x_scaler.fit_transform(X_train)x_test_sca = x_scaler.transform(X_test)y_train_sca = y_scaler.fit_transform(pd.DataFrame(y_train))

6.2 零均值规范化

对原始数据进行线性变换，经过处理的数据的均值为0，标准差为1。计算方式是将特征值减去均值，除以标准差。计算公式为：x* = (x-x.mean)/σ

from sklearn.preprocessing import StandardScaler#一般把train和test集放在一起做标准化，或者在train集上做标准化后，用同样的标准化器去标准化test集scaler = StandardScaler()train = scaler.fit_transform(train)test = scaler.transform(test)

作者：GC_AIDM

https://www.cnblogs.com/shenggang/p/12133278.html

—END—

hych666」，欢迎添加我的微信，更多精彩，尽在我的朋友圈 。

↓