首页 > 文章列表 > Python 3.x 中如何使用csv模块读取和写入CSV文件

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

csv 读取 写入
304 2023-08-05

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文件有所帮助。祝学习愉快!