【Python入门教程】第73篇 写入CSV文件

本文详细介绍了如何使用Python内置的csv模块将数据写入CSV文件,包括打开文件、创建CSVwriter对象、写入数据行以及处理非ASCII编码字符的方法。示例代码展示了如何一次性写入多行数据以及使用DictWriter类处理字典数据。通过设置newline参数,可以避免写入数据间出现额外的空白行。
摘要由CSDN通过智能技术生成

本篇我们介绍如何使用 Python 内置的 csv 模块将数据写入 CSV 文件。

写入 CSV 文件

在 Python 代码中写入 CSV 文件的步骤如下:

  • 首先,使用内置的 open() 函数以写入模式打开文件。
  • 其次,调用 writer() 函数创建一个 CSV writer 对象。
  • 然后,利用 CSV writer 对象的 writerow() 或者 writerows() 方法将数据写入文件。
  • 最后,关闭文件。

以下代码实现了上面的步骤:

import csv

# open the file in the write mode
f = open('path/to/csv_file', 'w')

# create the csv writer
writer = csv.writer(f)

# write a row to the csv file
writer.writerow(row)

# close the file
f.close()

使用 with 语句可以避免调用 close() 方法关闭文件,从而使得代码更加精简:

import csv

# open the file in the write mode
with open('path/to/csv_file', 'w') as f:
    # create the csv writer
    writer = csv.writer(f)

    # write a row to the csv file
    writer.writerow(row)

如果数据中包含非 ASCII 编码字符,需要在 open() 函数中指定字符编码。以下示例演示了如何将 UTF-8 字符写入 CSV 文件:

import csv

# open the file in the write mode
with open('path/to/csv_file', 'w', encoding='UTF8') as f:
    # create the csv writer
    writer = csv.writer(f)

    # write a row to the csv file
    writer.writerow(row)

示例

以下示例演示了如何将数据写入 CSV 文件:

import csv  

header = ['id', 'stu_id', 'course_name', 'course_score']
data = [1, 1, 'English', 100]

with open('score.csv', 'w', encoding='UTF8') as f:
    writer = csv.writer(f)
    # write the header
    writer.writerow(header)
    # write the data
    writer.writerow(data)

如果打开 score.csv 文件,我们会发现一个问题,就是两行数据之间存在一个额外的空白行:

在这里插入图片描述
为了删除多余的空白行,我们可以为 open() 函数指定关键字参数 newline=’’:

import csv  

header = ['id', 'stu_id', 'course_name', 'course_score']
data = [1, 1, 'English', 100]

with open('score.csv', 'w', encoding='UTF8', newline='') as f:
    writer = csv.writer(f)
    # write the header
    writer.writerow(header)
    # write the data
    writer.writerow(data)

再次打开 score.csv 文件,显示的内容如下:

在这里插入图片描述

写入多行数据

如果想要一次写入多行数据,我们可以使用 writerows() 方法。例如:

import csv  

header = ['id', 'stu_id', 'course_name', 'course_score']
data = [
    [1, 1, 'English', 100],
    [2, 1, 'Math', 95],
    [3, 2, 'English', 96]
]

with open('score.csv', 'w', encoding='UTF8', newline='') as f:
    writer = csv.writer(f)
    # write the header
    writer.writerow(header)
    # write the data
    writer.writerows(data)

新文件的内容如下:

在这里插入图片描述

DictWriter 类

如果写入的数据行由字典组成,我们可以使用 csv 模块 的DictWriter 类将数据写入文件。例如:

import csv

# csv header
fieldnames = ['id', 'stu_id', 'course_name', 'course_score']

# csv data
rows = [
    {'id': 1,
    'stu_id': 1,
    'course_name': 'English',
    'course_score': 100},
    {'id': 2,
    'stu_id': 1,
    'course_name': 'Math',
    'course_score': 95},
    {'id': 3,
    'stu_id': 2,
    'course_name': 'English',
    'course_score': 96}
]

with open('score.csv', 'w', encoding='UTF8', newline='') as f:
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(rows)

以上代码的执行过程如下:

  • 首先,定义两个存储字段名和数据行的变量。
  • 其次,调用 open() 函数以写入模式打开 CSV 文件。
  • 然后,创建一个新的 DictWriter 类实例。
  • 接着,调用 writeheader() 方法写入标题行。
  • 最后,使用 writerows() 方法写入数据行。

总结

  • csv.writer() 函数和 DictWriter 类都可以将数据写入 CSV 文件。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不剪发的Tony老师

为 5 个 C 币而折腰。

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值