Python读写CSV文件

1 csv文件简介

逗号分隔符Comma-Separated Values

csv是一个被行分隔符、列分隔符划分成行和列的文本文件
CSV不指定字符编码
行分隔符为\r\n,最后一行可以没有换行符
列分隔符常为逗号或制表符
每一行称为一条记录record
字段可以使用双引号括起来，也可以不适用，如果字段中出现双引号，逗号，换行符必须用双引号括起来
表头可选，和字段列对齐就行了

csv 库中有4个常用的对象：

• csv.reader：以列表的形式返回读取的数据。
• csv.writer：以列表的形式写入数据。
• csv.DictReader：以字典的形式返回读取的数据。
• csv.DictWriter：以字典的形式写入数据。

关于写入，需要注意：

• 在打开文件时，需要添加newline = ""。否则，会在每2行有效内容之间添加一行空白。
• 如果要保存的内容有中文，而且之后需要用Excel打开文件，那么需要选用utf-8-sig编码。如果使用utf-8编码，会导致使用Excel查看文件时中文乱码。

2 writer对象

写入一行或多行数据

import csv

csv_file = "test.csv"
headers = ["Number", "name", "age", "city"]
data = [
    [1, "zhangsan", 18, 'bei"ji"ng'],
    [2, "lisi", 20, "shang'hai"],
    [3, "wanger", 22, "shenzhen"],
]

with open(csv_file, "w", encoding="utf-8", newline="") as fp:
    csv_fp = csv.writer(fp)
    csv_fp.writerow(headers)
    csv_fp.writerows(data)

3 DictWriter对象

在写入字典序列类型数据的时候，需要传入两个参数，一个是文件对象——f，一个是字段名称——fieldnames，到时候要写入表头的时候，只需要调用writerheader方法，写入一行字典系列数据调用writerrow方法，并传入相应字典参数，写入多行调用writerows

import csv

csv_file = "test.csv"
headers = ["name", "age", "city"]
data = [
    {"name": "zhangsan", "age": 18},
    {"name": "lisi", "age": 20, "city": "shenzhen"},
]

with open(csv_file, "w", newline="") as fp:
    csv_fp = csv.DictWriter(fp, fieldnames=headers)
    csv_fp.writeheader()
    csv_fp.writerows(data)

4 reader读取数据

读取数据

import csv

csv_file = "test.csv"

with open(csv_file, "r") as fp:
    csv_fp = csv.reader(fp)
    for row in csv_fp:
        print(row)  # row[0]

5 DictReader读取数据