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