使用 csv 模块如何读取 csv 中的特定行 - Python

CSV(Comma-Separated Values)是一种常用的数据格式，由于其结构简单、易于存储和处理，被广泛应用于数据库、Excel等各种数据源之间的数据交换。在Python中，我们可以使用csv模块来处理CSV文件。

如何读取CSV文件

读取CSV文件的步骤很简单，可以分为如下几个步骤：

1. 导入csv模块
2. 使用open()打开CSV文件
3. 使用csv.reader()读取CSV文件
4. 遍历CSV文件中的每一行数据

接下来，我们就使用一个简单的示例来说明如何读取CSV文件。

示例代码

import csv

filename = 'data.csv'

# 使用with语句打开CSV文件
with open(filename, 'r') as csvfile:
    # 创建csv.reader对象
    csvreader = csv.reader(csvfile)
    
    # 遍历CSV文件中的每一行数据
    for row in csvreader:
        print(row)

示例输出

['Name', 'Age', 'Gender']
['Tom', '20', 'Male']
['Jerry', '22', 'Male']
['Lucy', '18', 'Female']
['Lily', '19', 'Female']

如何读取特定行

如果你只需要读取CSV文件中的特定行数据，可以使用csv.reader()对象的__next__()方法一行一行地读取数据，当读到目标行时就可以停止读取。

示例代码

import csv

filename = 'data.csv'

# 使用with语句打开CSV文件
with open(filename, 'r') as csvfile:
    # 创建csv.reader对象
    csvreader = csv.reader(csvfile)
    
    # 循环遍历CSV文件中的每一行数据，直到读到目标行
    for i in range(3):  # 读取第4行（从0开始计数）
        row = csvreader.__next__()
    
    # 输出目标行数据
    print(row)

示例输出

['Lucy', '18', 'Female']

在上面的示例中，我们使用csvreader.__next__()方法读取CSV文件中的每一行数据，当读到第4行数据时就保存在row变量中，最后输出了该行数据。

总结

通过本文的介绍，我们可以知道Python的csv模块提供了一组操作CSV文件的API，使用csv模块可以方便地读写CSV文件，实现数据的快速导入和导出。对于我们常见的数据处理、数据分析等应用场景，使用Python对CSV文件进行读写、数据清洗、统计计算等操作是非常方便、快捷、高效的。