关于Python的学习记录(二十二_读写 CSV 文件)

CSV 文件介绍

CSV（Comma Separated Values）全称逗号分隔值文件是一种简单、通用的文件格式，被广泛的应用于应用程序（数据库、电子表格等）数据的导入和导出以及异构系统之间的数据交换。因为 CSV 是纯文本文件，不管是什么操作系统和编程语言都是可以处理纯文本的，而且很多编程语言中都提供了对读写 CSV 文件的支持，因此 CSV 格式在数据处理和数据科学中被广泛应用。

CSV 文件有以下特点：

1. 纯文本；
2. 由一条条的记录组成（典型的是每行一条记录）；
3. 每条记录被分隔符（如逗号、分号、制表符等）分隔为字段（列）；
4. 每条记录都有同样的字段序列。

CSV 文件可以使用文本编辑器或类似于 Excel 电子表格这类工具打开和编辑，当使用 Excel 这类电子表格打开 CSV 文件时，你甚至感觉不到 CSV 和 Excel 文件的区别。很多数据库系统都支持将数据导出到 CSV 文件中，当然也支持从 CSV 文件中读入数据保存到数据库中。

将数据写入 CSV 文件

现有五个学生三门课程的考试成绩需要保存到一个 CSV 文件中，要达成这个目标，可以使用 Python 标准库中的csv模块，该模块的writer函数会返回一个csvwriter对象，通过该对象的writerow或writerows方法就可以将数据写入到 CSV 文件中，具体的代码如下所示。

import csv
import random

with open('scores.csv', 'w') as file:
    writer = csv.writer(file)
    # 先写入第一行：列名
    writer.writerow(['姓名', '语文', '数学', '英语'])
    names = ['关羽', '张飞', '赵云', '马超', '黄忠']
    for name in names:
        scores = [random.randint(50, 101) for _ in range(3)]
        scores.insert(0, name)
        writer.writerow(scores)
# 将 csv 文件内容打印出来
with open('scores.csv', 'r') as file:
    print(file.read())

生成的 CSV 文件的内容如下：

姓名,语文,数学,英语
关羽,98,86,61
张飞,86,58,80
赵云,95,73,70
马超,83,97,55
黄忠,61,54,87

写文件的时候所使用的 writer 函数，除了传入要写入数据的文件对象外，还可以通过delimiter、quotechar、quoting参数来指定分隔符（默认是逗号）、包围值的字符（默认是双引号）以及包围的方式。其中，包围值的字符主要用于当字段中有特殊符号时，通过添加包围值的字符可以避免二义性。

writer = csv.writer(file, delimiter='|', quoting=csv.QUOTE_ALL)

生成的 CSV 文件的内容。

"姓名"|"语文"|"数学"|"英语"
"关羽"|"88"|"64"|"65"
"张飞"|"76"|"93"|"79"
"赵云"|"78"|"55"|"76"
"马超"|"72"|"77"|"68"
"黄忠"|"70"|"72"|"51"

从 CSV 文件读取数据

通过csv模块的reader函数创建出csvreader对象，该对象是一个迭代器，可以通过for-in循环读取到文件中的数据。

import csv
import random
with open('scores.csv', 'w') as file:
    writer = csv.writer(file)
    # 先写入第一行：列名
    writer.writerow(['姓名', '语文', '数学', '英语'])
    names = ['关羽', '张飞', '赵云', '马超', '黄忠']
    for name in names:
        #列表推导式，它的作用是生成一个包含三个随机整数的列表
        scores = [random.randint(50, 101) for _ in range(3)]
        scores.insert(0, name)
        writer.writerow(scores)

with open('scores.csv', 'r') as file:
    reader = csv.reader(file)
    # data_list 是一个列表对象，该列表对象包含了一行中所有的字段。
    for data_list in reader:
        # 我们会给读到的数据打印出行信息
        print(reader.line_num, end='\t')
        for elem in data_list:
            # 打印行里面的每一个字段
            print(elem, end='\t')
        print()