csv也是一种和Excel类似的表格类型文件,不过它里面放的一般都是不需要经常去进行改动的数据,而在python内也会有需要要将数据保存为csv文件。那么本篇文章就要来给大家分享一下python保存csv文件的方法,感兴趣的话就和小编一起往下阅读吧。

Python怎么保存csv文件?Python保存csv格式文件需要用什么方法

1.虽然python中有内置模块能够操作csv文件,不过pandas这个第三方库在处理速度和代码编写方面都是优于内置模块了。同时也需要将os模块导入进来判断一下保存csv文件的路径是否存在或者有同名文件,代码示例如下所示:

import os
import pandas as pd

2.模块导入完成之后就是要用变量来保存csv文件路径了,然后定义两个空集合,作用是保存列数据。再使用listdir()方法将文件路径里面的所有文件和文件夹都获取到并用for循环迭代,示例如下:

path = 'data/train/'
img_list=[]
lable_list=[]
testList = os.listdir(path)
for file in testList:

3.在for循环代码块里面向一个列表对象内添加文件名,然后定义一个参数并赋值为aa,然后将这个变量添加到另一个列表中。循环迭代完毕之后这两个列表内就有数据了,示例如下:

img_list.append(file)
label='aa'
lable_list.append(label)

4.第四步也就是最后一步了,调用zip()函数将这两个列表合并在一起并且使用list()方法再次转为列表。然后用pandas库中的DataFrame()方法将data参数表示的数据源设置为该列表对象,并指定两个列名。最后用to_csv()方法将其保存起来即可,详细代码如下所示:

img_label_list2 = list(zip(img_list, lable_list))
df3 = pd.DataFrame(data=img_label_list2,columns=['filepath', 'label'])
df3.to_csv('result.csv',index=False)

以上就是关于“Python怎么保存csv文件?Python保存csv格式文件需要用什么方法”的全部内容了,希望对你有所帮助。

更多python相关文章请访问分类:python

【版权声明】本文图文出自大盘站@dapan.cc,转载请注明出处!