1.csv文件写入:
写入单行csv文件:
import csv
test = [1, 2, 'aa']
with open('egg.csv', 'w') as file:
csvwriter = csv.writer(file, lineterminator='\n')
csvwriter.writerow(test)
写入多行csv文件:
import csv
test = [[1, 2], [3, 4, 'c'], ['a']]
with open('egg.csv', 'w') as file:
csvwriter = csv.writer(file, lineterminator='\n')
csvwriter.writerows(test)
注意!!!writer函数中的参数lineterminator=’\n’非常重要,否则会出现产生的csv文件中有多余的空行情况
使用writerows函数最大的优势是进行写入csv中的格式,和我们在python中list或者np.array中的格式完全一样,在写入的时候不需要写循环或者其他操作,写入过程非常直观。
当我们只写一行到csv文件时,最好是用writerow函数。当然如果写入的list中都是str类型数据(a = [’s’, ‘a’])时,用writerows写入也不会报错。但是如果写入的list中有int类型数据(a = [‘a’ ,’d’, 2]),用writerows函数就会报错。
2.csv文件读取
import csv
with open('egg.csv', 'r') as file:
csv_data = csv.reader(file)
for i, csv_name in enumerate(csv_data):
if i == 0:
csv_str = [[x for x in csv_name]]
else:
csv_str = csv_str + [[x for x in csv_name]]
csv文件进行读取的时候首先会将一行数据读进来,就是for i, csv_name in enumerate(csv_data),但是读进来的这一行数据我们是不能直接读取的,需要我们通过循环[x for x in csv_name]将这一行数据进行读取。
该函数读进来的格式和原来csv中的格式相同。