Pandas——如何提取/拆分csv文件中的指定列

在机器学习中，有时会需要将数据集的某个特征或多个特征提取出来，以便进行训练或对比，下面记录一种拆分csv文件中列（一般为特征）的方法。

'''
以下代码实现功能包括：
1.将'train.csv'文件中的'Id'、'SalePrice'特征提取到'train_price.csv'
2.将'train.csv'文件中除'SalePrice'的特征（包括'Id'特征）提取到'train_no_price.csv'
'''
'''
此代码用到的'train.csv'为某地区房屋基本信息及销售信息等，为方便学习，此文件全部内容已分享在文章末尾
'''
import pandas as pd
train = pd.read_csv('train.csv')#读取训练集
train.iloc[:,:].to_csv('train_price.csv',
				mode = 'w',
				index_label = 'Id',
				index = None#不添加新索引，若此处不设为None，会自动生成一列从0开始的索引
				)#将'train.csv'复制到'train_price.csv'
train_price = pd.read_csv('train_price.csv')
train_price.drop(train_price.iloc[:,1:-1],
				axis = 1,
				inplace=True#对原始对象进行修改
				)#删除特征'Id'及'SalePrice'以外的所有特征
train_price.iloc[:,:].to_csv('train_price.csv',
				mode = 'w',
				index_label = 'Id',
				header = True,
				index = None
				)#以只包含'Id'及'SalePrice'特征的csv文件覆盖原文件
train.iloc[:,0:-1].to_csv('train_no_price.csv',
				mode = 'w',
				index_label = 'Id',
				header = True,
				index = None
				)#将'train.csv'文件中除'SalePrice'的特征（包括'Id'特征）提取

点击此处下载train.csv，提取码：gszw