[Pandas]——csv存储注意事项

处理数据得到结果时，很多时候我们会保存为csv格式文件，这里就记录下存储相关的内容

一、csv文件的两种存储方式

1、直接存储

对存储后的list数据先转成DataFrame格式进行直接存储

#一次性存储到csv文件
def save2csv(position, tensor_list, filename):
    total_list = [[info[0], info[1]] for info in zip(position, tensor_list)]
    df = pd.DataFrame(columns=['cell_name', 'tensor_info'], data=total_list)
    df.to_csv(filename, index=False, encoding='utf-8')

2、按行存储

开始的时候，我就是按照下面的方式进行存储的，但是会发现列表名多次存储在csv文件中

上面的问题是因为：没有加header参数，该参数设置为0，就会忽略列名，这样就不会出现列名多次存储的现象

二、一些重要参数说明

• index参数：是否保留行索引
• header参数：是否保留列名
• columns参数：保存哪些列，eg：columns=['col1', 'col3']
• encoding参数：指定参数文件的编码格式
• sep参数：指定分隔符，默认是逗号
• mode参数：a追加，默认是w写入模式