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)