Python函数实现CSV文件读取和写入

在Python中，我们可以使用csv模块来读取和写入CSV文件。CSV文件是一种常见的文件格式，用于存储和传输表格数据。

**CSV文件的读取：**

首先，我们需要导入csv模块。然后，使用open函数打开CSV文件，并以只读模式打开它。

import csv

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上述代码中，我们使用了csv.reader()函数来创建一个csv.reader对象，通过该对象的reader()方法可以逐行读取CSV文件的内容。

你也可以通过循环遍历reader对象获取每一行的数据。每一行数据以列表的形式返回。

如果你的CSV文件中包含头部信息，可以使用next()函数跳过第一行。例如：

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    headers = next(reader)
    for row in reader:
        print(row)

这样就可以将数据从CSV文件中读取出来。

**CSV文件的写入：**

写入CSV文件也非常简单。首先创建一个csv.writer对象，并使用open函数以写入模式打开CSV文件。

import csv

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

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

在这个例子中，我们使用一个包含表头的data列表来写入CSV文件。使用writer.writerow()方法将每一行写入CSV文件中。注意要在open()函数中设置newline=''参数，这样可以避免写入CSV文件时生成额外的空行。

以上就是使用Python函数实现CSV文件的读取和写入的简单介绍。csv模块还有很多其他的函数和方法，可以根据需要进行进一步的学习和使用。