如何使用Python函数编写和读取CSV文件

CSV（逗号分隔值）文件是一种以逗号分隔数据字段的文本文件格式。Python中有许多方法可以读取和编写CSV文件。在本文中，我将介绍如何使用Python函数编写和读取CSV文件。

编写CSV文件：

要编写CSV文件，我们首先需要导入Python的csv模块。然后，我们可以使用csv.writer()函数创建一个写入对象，该对象允许我们将数据行写入CSV文件。

下面是一个简单的示例，演示如何将数据行写入CSV文件：

import csv

# 要写入的数据
data = [['姓名', '年龄', '性别'], ['张三', '25', '男'], ['李四', '30', '女'], ['王五', '35', '男']]

# 打开CSV文件进行写入
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

print('写入完成')

在上面的示例中，我们首先创建了一个包含数据行的列表。然后，我们使用open()函数打开一个名为data.csv的文件，并将其指定为写入模式。然后我们创建了一个csv.writer()对象，该对象将数据写入文件中。最后，我们使用writer.writerows()函数将整个数据列表写入CSV文件。最后，我们打印了一条消息，以指示数据写入完成。

读取CSV文件：

要读取CSV文件，我们可以使用Python的csv模块中的csv.reader()函数创建一个读取对象。然后，我们可以使用该对象从CSV文件中逐行读取数据。

下面是一个简单的示例，演示如何从CSV文件中读取数据：

import csv

# 打开CSV文件进行读取
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

print('读取完成')

在上面的示例中，我们首先使用open()函数打开名为data.csv的文件，并将其指定为读取模式。然后，我们使用csv.reader()函数创建一个读取对象，该对象用于从CSV文件中读取数据。接下来，我们使用一个循环迭代读取器对象，打印每一行的数据。最后，我们打印了一条消息，以指示数据读取完成。

除了逐行读取数据外，您还可以将数据加载到列表中：

import csv

# 打开CSV文件进行读取
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

print(data)
print('读取完成')

在上面的示例中，我们将读取器对象转换为列表，并将其保存在data变量中。然后我们打印了整个数据列表。最后，我们打印了一条消息，以指示数据读取完成。

综上所述，使用Python的csv模块编写和读取CSV文件非常简单。无论是编写数据还是读取数据，您都可以使用csv.writer()和csv.reader()函数来处理CSV文件。