python基础教程[3]:csv文件写入和读取

来源:互联网 发布:淘宝客服怎么做兼职 编辑:程序博客网 时间:2024/05/01 19:00

1.csv文件写入:
写入单行csv文件:

import csvtest = [1, 2, 'aa']with open('egg.csv', 'w') as file:    csvwriter = csv.writer(file, lineterminator='\n')    csvwriter.writerow(test)

写入多行csv文件:

import csvtest = [[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 csvwith 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中的格式相同。
这里写图片描述

原创粉丝点击