欢迎访问宙启技术站
智能推送

使用Python的write()函数写入CSV格式文件

发布时间:2023-12-17 12:00:49

CSV(Comma Separated Values)是一种常用的数据格式,用于将表格数据存储为纯文本文件。Python的写入CSV文件的函数是write()函数。

使用write()函数写入CSV文件的基本步骤如下:

1. 打开要写入的CSV文件,可以使用内置的open()函数。可以指定文件名、打开模式和编码。打开模式可以是写入模式('w'),如果文件不存在则创建,如果文件存在则覆盖。如果要追加内容到已有文件,则可以使用追加模式('a')。

2. 创建一个csv.writer对象,用于写入数据到CSV文件。可以使用内置的csv模块的writer()函数,传入文件对象和其他参数(如分隔符、引用符等)。

3. 使用writerow()方法向文件中写入一行数据,参数是一个列表或元组,每个元素对应一个CSV文件的单元格。可以重复调用writerow()方法,写入多行数据。

4. 关闭CSV文件,释放资源,可以使用文件对象的close()方法或者with语句。

下面是一个使用write()函数写入CSV文件的例子:

import csv

# 打开或创建CSV文件
with open('data.csv', 'w', encoding='utf-8') as file:
    # 创建csv.writer对象
    writer = csv.writer(file)

    # 写入表头
    writer.writerow(['姓名', '年龄', '性别'])

    # 写入数据行
    writer.writerow(['张三', 20, '男'])
    writer.writerow(['李四', 25, '女'])
    writer.writerow(['王五', 30, '男'])

# 文件关闭后,数据已写入CSV文件

在上面的例子中,首先使用open()函数创建一个名为data.csv的CSV文件,并指定打开模式为写入模式。然后使用csv.writer()函数创建一个csv.writer对象,用于写入数据到该文件。然后使用writerow()方法先写入表头,再写入数据行。最后,使用with语句来自动关闭文件,这样可以确保在文件关闭后数据会被写入到文件中。

以上就是使用Python的write()函数写入CSV格式文件的基本方法和例子。你可以根据需要来修改和扩展这个例子,来适应你的具体需求。