使用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文件中。
