Python中使用writer()函数生成CSV文件并添加新行
发布时间:2024-01-13 18:14:19
在Python中,可以使用csv.writer()函数生成CSV文件并添加新行。CSV文件是逗号分隔的文本文件,在其中每个字段占一列,每行以换行符结束。下面是一个简单的例子,演示了如何使用csv.writer()函数生成CSV文件并添加新行:
import csv
# 创建CSV文件
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
# 写入列名
writer.writerow(['Name', 'Age', 'Country'])
# 写入一行数据
writer.writerow(['John', 28, 'USA'])
# 写入多行数据
rows = [
['Alice', 32, 'Canada'],
['Bob', 24, 'UK'],
['Chris', 37, 'Australia']
]
writer.writerows(rows)
上面的代码首先导入了csv模块,并使用open()函数创建一个名为"data.csv"的文件。我们在打开文件时使用'w'模式,表示创建一个新文件或覆盖现有文件。newline=''参数确保在写入文件时不会出现额外的空行。
然后,我们使用csv.writer()函数创建了一个writer对象,并将其赋值给writer变量。
接下来,我们使用writer.writerow()方法写入了一行列名。在这个例子中,我们以列表的形式提供了列名。
然后,我们使用writer.writerow()方法一次写入了一行数据。我们使用列表提供了姓名、年龄和国家。
最后,我们使用writer.writerows()方法一次写入了多行数据。我们以二维列表的形式提供了多行数据。
执行上面的代码后,将生成一个名为"data.csv"的文件。文件内容如下:
Name,Age,Country John,28,USA Alice,32,Canada Bob,24,UK Chris,37,Australia
这是一个标准的CSV文件,其中每个逗号分隔的值占一列,每行以换行符结束。
使用csv.writer()函数,我们可以轻松地生成和更新CSV文件。可以根据需要将新行添加到CSV文件,提供各种数据类型作为列的值。这使得CSV文件成为处理和存储简单数据的一种非常方便的格式。
