处理数据得到结果时,很多时候我们会保存为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写入模式
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)