Python-将数据写入CSV文件
时间:2020-02-23 14:43:40 来源:igfitidea点击:
在本教程中,我们将学习如何使用Python将数据写入CSV文件。
在上一教程中,我们介绍了如何从CSV文件读取数据。
导入csv
模块
为了将数据保存在CSV文件中,我们必须借助csv
模块。
因此,继续通过编写以下代码来导入模块。
import csv
将数据写入CSV文件
要将数据写入CSV文件,我们将首先使用open()
方法创建一个新文件。
让我们将文件命名为" sample.csv"。
要写入数据,我们将使用csv
模块的DictWriter
类。
我们将传递包含对新创建的CSV文件的引用的文件对象以及包含字段名称的变量。
为了写出字段名,我们将使用writeheader()
方法。
为了写数据行,我们将使用writerow()
方法。
示例数据
在本教程中,我们将在" sample.csv"文件中写入以下数据。
playerid,score 1,10 2,9 3,12
其中,第一行的" playerid,score"是字段。
其余3行是数据。
使用字典写数据
在下面的Python程序中,我们将示例数据写入" sample.csv"文件中。
数据以字典形式存储在列表中。
# import module import csv try: # create file fobj = open('score.csv', 'w') # fields fields = ['playerid', 'score'] # data to save in csv file data = [ {'playerid': 1, 'score': 10}, {'playerid': 2, 'score': 9}, {'playerid': 3, 'score': 12} ] # writer writer = csv.DictWriter(fobj, fields) # write the header writer.writeheader() # write the data for row in data: writer.writerow(row) except: print("An error occurred while writing the file.") finally: fobj.close()
执行上述代码后,我们将获得包含以下数据的sample.csv
文件。
playerid,score 1,10 2,9 3,12