Python 3.x 中如何使用csv模块读取和写入CSV文件

Python 3.x 中如何使用csv模块读取和写入CSV文件

引言：
CSV（Comma-Separated Values）是一种常见的文件格式，用于存储表格数据。在Python中，可以使用csv模块来处理CSV文件，该模块提供了读取和写入CSV文件的功能。本文将介绍如何使用csv模块在Python 3.x中读取和写入CSV文件，并提供代码示例。

一、使用csv模块读取CSV文件

首先，我们需要导入csv模块：

import csv

接下来，我们可以使用csv.reader()函数来读取CSV文件。该函数需要传入一个文件对象，我们可以使用open()函数来打开一个CSV文件，并把文件对象作为参数传给csv.reader()函数。

假设我们有一个名为data.csv的CSV文件，该文件的内容如下：

name,age,city
John,25,New York
Peter,30,Chicago
Sarah,28,Los Angeles

下面是一个简单的代码示例，展示如何使用csv模块来读取CSV文件：

import csv

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

运行上述代码，将会输出CSV文件中的每一行数据。在这个示例中，我们使用了一个for循环来遍历csv_reader对象，以逐行打印CSV文件中的数据。

二、使用csv模块写入CSV文件

与读取CSV文件类似，我们也可以使用csv模块来写入CSV文件。同样需要导入csv模块：

import csv

然后，我们可以使用csv.writer()函数来创建一个csv_writer对象，该对象可以将数据写入CSV文件。该函数需要传入一个文件对象，同样可以使用open()函数来打开一个CSV文件，并把文件对象作为参数传给csv.writer()函数。

下面是一个简单的代码示例，展示如何使用csv模块来写入CSV文件：

import csv

data = [
    ['name', 'age', 'city'],
    ['John', '25', 'New York'],
    ['Peter', '30', 'Chicago'],
    ['Sarah', '28', 'Los Angeles']
]

with open('output.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

运行上述代码，将会在当前目录下创建一个名为output.csv的CSV文件，并将data中的数据写入到该文件中。

在代码示例中，我们首先定义了一个二维列表data，其中包含了CSV文件的每一行数据。然后，我们使用csv.writerows()函数将整个列表写入CSV文件。

总结：
通过csv模块，我们可以很方便地在Python 3.x中读取和写入CSV文件。在读取CSV文件时，我们可以使用csv.reader()函数和for循环来逐行处理数据；在写入CSV文件时，我们可以使用csv.writer()函数和csv_writer.writerows()函数来将数据写入CSV文件中。

希望本文对你在Python中使用csv模块读取和写入CSV文件有所帮助。祝学习愉快！

以上是Python 3.x 中如何使用csv模块读取和写入CSV文件的详细内容。更多信息请关注PHP中文网其他相关文章！