Python CSV模块使用实例

当我们需要从CSV文件中读取或写入数据时，Python提供了一个内置的CSV模块，该模块可以轻松地读取和写入CSV文件。接下来就让我们来详细讲解一下Python CSV模块的使用。

CSV模块的导入

要使用CSV模块，我们需要先将其导入到Python脚本中。代码如下：

import csv

读取CSV文件

要读取CSV文件，需要使用Python内置的csv.reader()函数。在使用该函数时，我们需要指定CSV文件的路径和分隔符。示例代码如下：

import csv

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

在上面的代码中，我们首先打开了一个名为data.csv的CSV文件，然后使用csv.reader()函数创建了一个读取器对象。然后，我们可以使用for循环从读取器对象中迭代每一行数据。

写入CSV文件

要写入CSV文件，需要使用Python内置的csv.writer()函数。在使用该函数时，我们需要指定CSV文件的路径和分隔符。示例代码如下：

import csv

data = [
    ['name', 'age', 'city'],
    ['Alice', '20', 'New York'],
    ['Bob', '25', 'San Francisco'],
    ['Charlie', '30', 'London']
]

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

在上面的代码中，我们首先定义了一个名为data的二维列表，每个列表代表CSV文件的一行。然后，我们打开了一个名为data.csv的CSV文件，并使用csv.writer()函数创建了一个写入器对象。然后，我们可以使用writer.writerows()函数将整个列表写入CSV文件中。

示例说明

示例一

假设我们有一个CSV文件data.csv，其中包含着一些学生的信息，格式如下：

name,age,city
Alice,20,New York
Bob,25,San Francisco
Charlie,30,London

现在，我们可以使用以下代码来读取CSV文件中的每一行数据：

import csv

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

输出结果如下：

['name', 'age', 'city']
['Alice', '20', 'New York']
['Bob', '25', 'San Francisco']
['Charlie', '30', 'London']

示例二

假设我们有一个名为data.csv的CSV文件，我们想要将以下数据写入到该文件中：

name,age,city
Alice,20,New York
Bob,25,San Francisco
Charlie,30,London

现在，我们可以使用以下代码将数据写入到CSV文件中：

import csv

data = [
    ['name', 'age', 'city'],
    ['Alice', '20', 'New York'],
    ['Bob', '25', 'San Francisco'],
    ['Charlie', '30', 'London']
]

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

这将数据成功写入到名为data.csv的CSV文件中。