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

