携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第26天,点击查看活动详情
前言
CSV
文件是具有高度兼容格式的简单电子表格,是带有表格数据的文本文件,是以逗号分隔(因此称为逗号分隔值,Comma-Separated Values
)的简单表格格式。 CSV
文件可以使用 Python
的标准库创建,并且可以被各种电子表格软件所读取和修改。
Python 创建 CSV 文件
为了创建 CSV
文件,只需要使用 Python
的标准库,本节中,我们将介绍如何使用 Python
创建 CSV
电子表格。
首先,导入 csv
模块:
>>> import csv
定义电子表格标题,其中包含数据的输入方式(例如第一列表示用户名,第二列表示电影名,第三列表示用户对电影的评分)和要存储的数据:
>>> header = ('User name', 'Movie name', 'rating')
>>> data = [('001', 'Star Wars', 7.8), ('002', 'Back to Future', 8.0), ('002', 'Batman', 7.7), ('003', 'Spy 007', 8.3)]
将数据写入 CSV
文件,保存 movies_rating.csv
文件:
>>> with open('movies_rating.csv', 'w', newline='') as f:
... movies = csv.writer(f)
... movies.writerow(header)
... for row in data:
... movies.writerow(row)
...
29
19
24
16
17
在以上代码中,我们首先使用写 (w
) 模式打开一个新文件 movies_rating.csv
。 csv
文件中的文件句柄对象引用该文件,csv.writer()
函数使用该文件创建 CSV
文件并进行写入操作,为了在写入完成后自动关闭文件句柄,我们使用 with
关键字将以上操作包含在同一代码块中,以便在操作完成后自动关闭文件。
open
方法中的 newline=''
参数是为了在写入文件时能够控制换行符格式并避免不兼容问题,例如可以防止添加多个换行符。在写入普通文件时,通常不会遇到此问题,但有时 CSV
格式可能要求换行符是特定字符,而不能使用默认字符。例如,如果要在不同操作系统上使用 CSV
文件,就可能会发生这种情况。因此,最好在 CSV
文件编写时中明确得进行声明。
在 writer
中我们使用 writerow()
方法逐行写入数据内容。第一行是 csv
标头 header
,其余行是数据行。
在电子表格软件中检查生成的 CSV
文件。在下图中可以看到,该文件使用 WPS
软件打开展示,与我们的预期效果一致: