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

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中的格式相同。