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读取数据
|