to_csv保存指定列的方法

df是DataFrame的数据，它的列为

['代码', '名称',  '最高', '最低']

现在我只想将‘代码’、“名称”两列内容存入csv，实现如下：

columns_to_save = ['代码', '名称']

df.代码 = df.代码.apply('="{}"'.format)#此行可以防止代码之前的0被忽略掉

df[columns_to_save].to_csv('result.csv', encoding='gbk', index=False)

下面举例：读取一个csv，精简其列，保存为另外一个csv。 

import pandas as pd# 读取CSV文件
data_base = pd.read_csv('base_info.csv', encoding='gbk')
print(type(data_base))
print(data_base.columns)
#<class 'pandas.core.frame.DataFrame'>
#Index(['代码', '名称', '列A', '列B', '列C'],dtype='object')columns_to_save = ['代码', '名称']
data_base.代码 = data_base.代码.apply('="{}"'.format)#此行可以防止代码之前的0被忽略掉
data_base[columns_to_save].to_csv('result.csv', encoding='gbk', index=False)