在Python 3中,你可以使用内置的`csv`模块来写入CSV文件。以下是一个简单的示例,展示了如何创建一个带有表头的CSV文件,并向其中写入数据:
import csv
定义数据和表头
data = [
["Name", "Age", "City"],
["Alice", 25, "New York"],
["Bob", 30, "San Francisco"],
["Charlie", 22, "Los Angeles"]
]
打开文件并创建csv写入对象
with open("output.csv", "w", newline="", encoding="utf-8") as csvfile:
csv_writer = csv.writer(csvfile)
写入表头
csv_writer.writerow(data)
写入数据行
csv_writer.writerows(data[1:])
这段代码首先定义了要写入的数据和表头,然后使用`with open`语句创建或打开一个名为`output.csv`的文件,并指定`newline=""`以避免在Windows系统中添加额外的空行。`csv.writer`对象被用来写入数据,首先写入表头,然后写入数据行。
如果你需要追加数据到已存在的CSV文件中,可以使用`a`(追加)模式打开文件:
import csv
定义要追加的数据
new_data = [
["David", 35, "Seattle"]
]
打开文件并创建csv写入对象
with open("output.csv", "a", newline="", encoding="utf-8") as csvfile:
csv_writer = csv.writer(csvfile)
写入数据行
csv_writer.writerows(new_data)
这段代码将在`output.csv`文件的末尾追加新的数据行。
请注意,如果你的数据中包含特殊字符,确保在打开文件时指定正确的编码,如`encoding="utf-8"`,以避免编码错误。