使用 Python 读写 csv 文件

非关系型数据库中，以文档型的数据库 MongoDB 最为著名。还有一个很好的全文检索引擎 Elasticsearch，基本上也可以当做一个文档型的数据库来使用。

创建 csv 文件：

import csvwith open("./csv_tutotial.csv", "a") as fw:    writer = csv.writer(fw)    writer.writerow(["c1", "c2", "c3"])    for x in range(10):        writer.writerow([x, chr(ord('a') + x), 'aaa'])

说明：1、这里“a”表示，打开文件的方式为“追加”；
2、打开这个 csv 文件可以看到，字符串没有用 “” 号括起来；

想为字符串增加双引号，可以这样写：
将

writer = csv.writer(fw)

替换成

writer = csv.writer(fw, quoting=csv.QUOTE_NONNUMERIC)

csv 文件可以使用的存储模式，请参考下面的这张表：

表 csv 文件可以使用的存储模式

csv 文件可以使用的存储模式 描述 QUOTE_ALL 为所有的栏增加双引号包围 QUOTE_MINIMAL 仅为包含特殊符号的栏增加双引号包围 QUOTE_NONNUMERIC 为所有非数字的栏增加双引号包围 QUOTE_NONE 在 reader() 函数中，表示不要去掉数据中的双引号包围

读取 csv 文件

读取 csv 文件：

from __future__ import print_functionimport csvwith open("./csv_tutotial.csv", "r") as fr:    # 这个方法返回的是迭代类型    rows = csv.reader(fr)    for row in rows:        print(row)

打印结果：

['c1', 'c2', 'c3']['0', 'a', 'aaa']['1', 'b', 'aaa']['2', 'c', 'aaa']['3', 'd', 'aaa']['4', 'e', 'aaa']['5', 'f', 'aaa']['6', 'g', 'aaa']['7', 'h', 'aaa']['8', 'i', 'aaa']['9', 'j', 'aaa']

观察可以发现：
1、原始的行数据按照逗号分隔被切分成了列表
2、Name 列中的双引号也被去掉了
在 Python 编程中，我们从来都不提倡“重复造轮子”。

假设要读取的文件内容是这样的：

c1|c2|c30|a|aaa1|b|aaa2|c|aaa3|d|aaa4|e|aaa5|f|aaa6|g|aaa7|h|aaa8|i|aaa9|j|aaa

我们就需要处理方言，默认情况下，分割符号是“,”，我们想把分隔符号改成“|”，这件事情就叫做 csv 的方言。
使用的方法，我们可以在创建 csv 文件的时候，“注册”一下，这个操作相当于“设置”。

方言指定了解析或写一个数据文件时使用的所有记号。

csv.register_dialect("pipes", delimiter='|')

完整代码：

import csvcsv.register_dialect("pipes", delimiter="|")with open("./csv_tutotial.csv", "r") as fr:    rows = csv.reader(fr, dialect="pipes")    for row in rows:        print(row)

当然，在我们 insert 到 csv 文件的时候，也可以这么做：

import csvcsv.register_dialect("pipes", delimiter="|")with open("./csv_tutotial.csv", "w") as fr:    writer = csv.writer(fr, dialect="pipes")    writer.writerow(['id', 'name', 'age'])    for x in range(10):        writer.writerow([x, chr(ord('a') + x), int(x) + 20])

总结：不论是在写 csv 文件还是在读 csv 文件的时候，一定要做两件事情：
（1）csv.register_dialect("pipes", delimiter="|")
（2）writer = csv.writer(fr, dialect="pipes")

将读取的内容转换为字典

将读取的内容转换为字典，使用的 csv 的 DictReader() 方法。

import csvwith open("./csv_tutotial.csv", 'r') as fr:    rows = csv.DictReader(fr)    for row in rows:        print(row)

带方言的 csv 文件读法：

import csvcsv.register_dialect("pipes", delimiter="|")with open("./csv_tutotial.csv", "r") as fr:    rows = csv.DictReader(fr, dialect="pipes")    for row in rows:        print(row['id'], row.get('name'), row['age'])

读取的结果是，第一行的永远作为了 key。

或者我们可以这样说，csv.DictReader() 方法读出来的数据，默认以第 1 行作为标记。

注意：如果在写入文件的时候，两行之间会空出一行来，则需要修改如下：

with open("./csv_tutotial.csv", "w",newline='') as fr: