Python新建CSV文件并写入数据

在Python编程中，CSV是一种非常常用的数据文件格式。CSV文件以纯文本形式存储数据，每行表示一条记录，每列表示不同的字段。本文将从以下多个方面详细介绍在Python中如何新建CSV文件并写入数据。

一、创建CSV文件

在Python中，要新建CSV文件可使用csv模块中的writer()函数来实现。以下是创建CSV文件的代码示例：

import csv 
 
with open('test.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '年龄', '性别'])
    writer.writerow(['张三', 18, '男'])
    writer.writerow(['李四', 21, '女'])

代码中，首先用import导入csv模块。使用open()函数来打开一个CSV文件（'test.csv'），并在文件中创建一个writer对象。如果CSV文件不存在，则会自动创建该文件。

当使用writerow()方法来写入数据时，每一个元素都会被编码为CSV中的字符串。

二、写入数据到CSV文件

通过writer对象的writerow()方法，可以向CSV文件中逐行写入数据。

下面的代码示例演示了如何向CSV文件中添加新的记录：

import csv
 
with open('test.csv', 'a', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['王五', 20, '男'])

在上面的代码中，使用了'w'打开文件模式来打开文件，表示需要覆盖原有的数据。使用'a'打开文件模式，则是追加写入新的数据。

三、写入多行数据到CSV文件

如果需要一次性写入多行数据，您可以通过writerows()方法实现。writerows()方法将迭代传递给它的任何可迭代对象，并将每个元素视为单独的行。

下面的代码示例演示了如何向CSV文件中添加多条记录：

import csv
 
with open('test.csv', 'a', newline='') as csvfile:
    writer = csv.writer(csvfile)
    rows = [['赵六', 19, '女'], ['钱七', 25, '男'], ['周八', 22, '男']]
    writer.writerows(rows)

在上面的代码中，rows被传递给writerows()方法，然后writerows()方法将所有的数据写入文件。

四、在CSV文件中写入字典数据

除了向CSV文件中添加列表外，您还可以向CSV文件中添加字典数据。下面是一个示例代码：

import csv
 
with open('test.csv', 'a', newline='') as csvfile:
    fieldnames = ['姓名', '年龄', '性别']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
 
    writer.writerow({'姓名': '孙九', '年龄': 28, '性别': '男'})
    writer.writerow({'姓名': '周十', '年龄': 23, '性别': '女'})

在上面的代码中，使用了DictWriter类实例化一个writer对象，并传递了一个fieldnames参数，用于指定CSV文件中的列名。

五、总结

本文从创建CSV文件、向CSV文件中逐行写入数据、写入多行数据到CSV文件、在CSV文件中写入字典数据这些方面，详细介绍了在Python中如何新建CSV文件并写入数据。以上例子可以帮助您更好地理解Python中的CSV文件操作，使您能够轻松地进行CSV文件的创建和修改。