您的位置:首页 > 编程语言 > Python开发

python学习随笔(1):读取与写入csv文件

2017-03-29 14:19 633 查看
在做机器学习项目的时候,很多时候样本的标签都是以.csv的格式给出来的,这也就需要对该标签文件进行读写,在python中,有一个专门处理csv的包,直接调用这个包,就可以使用其中很多的函数。

>>>import csv


读取csv文件的时候,可以使用csv.reader()函数,逐行读取csv文件,最后要将csv文件关闭,使用csv.close()函数,如下例。

#coding:utf-8
import csv
csvfile = file('csv_test.csv','rb')
reader = csv.reader(csvfile)
for line in reader:
print line
csvfile.close()


在上例中,csv,reader()函数将.csv文件的每一行以一个list的形式处理,每一个元素是一个字符串。

在将文件写入.csv时候,csv包提供了csv.writer()函数,并且提供了逐行写入writerow()和writerows()。

#coding:utf-8
import csv
csvfile = file('csv_test.csv','wb')
writer = csv.writer(csvfile)
data = [('img','label')]
writer.writerow(data)
csvfile.close()
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
相关文章推荐