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

使用write()函数在Python中向CSV文件写入数据的示例

发布时间:2023-12-13 01:25:32

在Python中,可以使用csv模块来读写CSV文件。该模块提供了writer()函数用于写入CSV文件。

下面是一个用write()函数向CSV文件写入数据的示例:

import csv

data = [
    ['Name', 'Age', 'City'],
    ['John', '25', 'New York'],
    ['Alice', '30', 'Los Angeles'],
    ['Bob', '35', 'Chicago']
]

# 打开CSV文件,以写入模式创建一个CSV写入器对象
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)

    # 使用CSV写入器写入数据
    writer.writerows(data)

print("数据已写入CSV文件")

在上面的示例中,首先创建了一个包含数据的列表data,每个子列表代表一行数据。然后,使用open()函数打开一个名为data.csv的文件,并指定写入模式和参数newline=''来避免在写入过程中出现额外的空行。

接下来,使用csv.writer()函数创建了一个CSV写入器对象writer,并将文件对象file作为参数传递给它。然后,使用writer.writerows()函数将数据写入CSV文件。writerows()函数可以同时写入多行数据。

最后,当所有数据都写入文件后,使用print()函数输出一条提示信息。

上述代码运行后,将会在当前目录下创建一个名为data.csv的CSV文件,并将数据写入该文件。

CSV文件内容如下:

Name,Age,City
John,25,New York
Alice,30,Los Angeles
Bob,35,Chicago

以上就是使用write()函数在Python中向CSV文件写入数据的示例。这个示例演示了如何使用csv模块来创建一个CSV写入器对象,并使用该对象将数据写入到CSV文件中。