python读写csv文件实例代码
Python读取与写入CSV文件需要导入Python自带的CSV模块,然后通过CSV模块中的函数csv.reader()与csv.writer()来进行CSV文件的读取与写入。
写入CSV文件
import csv # 需要import csv的文件包 out=open("aa.csv",'wb') # 注意这里如果以‘w'的形式打开,每次写入的数据中间就会多一个空行,所以要用‘wb' csv_write=csv.write(out,dialect='excel') # 下面进行具体的内容写入 ss=['a',3] csv_write.writerow(ss) #这样ss里面的信息就被写入到aa.csv文件中了,而且是在两列中 #这里如果需要写入多行,那么就采用循环进行循环输入就可以了 out.close()
读取CSV文件
import csv out=open("aa.csv",'r') read_csv=csv.reader(out,dialect='excel') for line in read_csv: #循环输出csv中的所有数据 print line out.close()
知识点扩展:
import csv
读文件
reader方法
with open('./data.csv', 'r') as f: reader = csv.reader(f) # python 2 header = reader.next() # python 3x 使用python内置的next header = next(reader) #获取真正的数据 for row in reader: print(row) # 打印每一行,列表类型,只能按位置索引
稍微介绍一下next(iterable, [default ])函数,它是python3x内置的,用于遍历可迭代对象的方法,其中iterable可以使字符串、列表、字典、元祖、集合等。若default给出,那么遍历到头时返回这个值,不然就抛异常。
这让我想到了字典的get()方法,它也有类似作用。比如,
# 如果dic里面没有key1这个键,那就返回我们指定的0 dic.get('key1', 0)
DictReader方法
写文件
跟读对应, 写操作也有两个方法
writer方法,适用列表的列表
with open('./data.csv', 'w') as f: header = ['col1', 'col2', ...] writer = csv.writer(f) writer.writeheader(header) for row in row_list: writer.writerow(row) # 或直接调用 writer.writerows(row_list)
DictWriter方法,适用字典列表
with open('./data.csv', 'w') as f: dictwriter = csv.DictWriter(f) dictwriter.writeheader(dic_list[0].keys()) for dic in dic_list: writer.writerow(dic) # 或者一次性写入,直接调用 wrier.writerows(dic_list)
都是些比较基础的操作,在这里记录一下。如果你觉得有帮助,就更好了。