CSV文件操作
CSV文件
- CSV简介:
- CSV文件通常用于我们在电子表格软件和纯文本之间交互数据,CSV文件内容是一些用逗号分隔的原始字符串。
- CSV文件的操作在Python中有单独的模块来使用,模块名为csv
CSV文本操作
- 导入模块:
import csv
- 打开文件对象:
fp = open(‘test.csv’, ‘a+’ )
- 返回专门用做csv处理的读,或写对象:
fp = csv.writer(fp);fp = csv.reader(fp)
CSV写文件
- csv提供了一个写函数:writerow()方法,为文件写入以逗号分割的数据 逗号分割的数据:常见的有list,tuple,set
writer = csv.writer(fp)
writer.writerow(('a','b','c','d'))
注意:在写csv文件的时候,可能会出现一行数据一个空行
可以通过在打开文件时指定: newline=’’ 或以二进制打开: open(‘data.csv’,’wb’)
open(‘data.csv’,’w’,newline=’’)
csvfile = open(‘data.csv’,’w’)
CSV读文件
reader = csv.reader(fp) #生成对应文件的csv读对象
for var1,var2,var3… in reader:
print(var1,var2,var3…)
字典形式读写文件
- 除了以上读写csv文件的方式,csv模块还提供了 csv.DictReader() 和 csv.DictWriter() 用于将字典形式的数据写入csv文件,以及从csv文件读取出的数据保存在字典中
import csv
#csv.DictWriter
with open('names.csv', 'w') as csvfile:
fieldnames = ['first_name', 'last_name']#指定标题列
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'first_name': 'Baked', 'last_name': 'Beans'})
writer.writerow({'first_name': 'Lovely', 'last_name': 'Spam'})
writer.writerow({'first_name': 'Wonderful', 'last_name': 'Spam'})
#csv.DictReader
with open('names.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
print(row['first_name'], row['last_name'])#根据标题列定义的数据取行内容