python 怎么读取csv的指定行数

原创

mob649e81607bf3 2024-01-31 07:31:41 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e81607bf3的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python读取CSV文件的指定行数

引言

在数据分析和处理的过程中，经常需要读取CSV文件，并针对其中的部分行进行处理。Python作为一种功能强大的编程语言，提供了多种方法来读取和处理CSV文件。本文将介绍如何使用Python读取CSV文件的指定行数，并提供一个实际问题的解决方案。

实际问题

假设我们有一个包含学生考试成绩的CSV文件，文件中的每一行代表一个学生的成绩记录，包括学生的姓名、学号和各科成绩。我们需要读取文件中的前n行数据，并进行一些数据分析和处理，例如计算平均成绩、查找成绩最高的学生等。

示例

首先，我们需要准备一个包含学生考试成绩的CSV文件。假设文件名为scores.csv，文件内容如下：

姓名,学号,语文,数学,英语
张三,001,90,95,85
李四,002,85,92,88
王五,003,92,88,90
赵六,004,88,90,95

接下来，我们使用Python读取CSV文件的指定行数，并进行一些数据处理操作。首先，我们需要导入csv模块，并定义一个函数read_csv_rows来读取指定行数的数据：

import csv

def read_csv_rows(filename, num_rows):
    rows = []
    with open(filename, 'r') as file:
        csv_reader = csv.reader(file)
        for i, row in enumerate(csv_reader):
            if i < num_rows:
                rows.append(row)
            else:
                break
    return rows

上述代码中，read_csv_rows函数接受两个参数：filename表示CSV文件的路径，num_rows表示要读取的行数。函数内部使用csv.reader来逐行读取文件内容，并将前num_rows行数据添加到rows列表中。当读取完指定行数的数据后，函数会立即返回rows列表。

接下来，我们可以调用以上函数来读取CSV文件的指定行数。例如，我们想读取文件中的前3行数据，可以使用以下代码：

filename = 'scores.csv'
num_rows = 3
rows = read_csv_rows(filename, num_rows)
print(rows)

运行以上代码，输出结果为：

[['姓名', '学号', '语文', '数学', '英语'], ['张三', '001', '90', '95', '85'], ['李四', '002', '85', '92', '88']]

在以上示例中，我们成功读取了CSV文件中的前3行数据，并将其存储在rows列表中。

数据处理

一旦我们成功读取了CSV文件的指定行数，就可以对这些数据进行进一步的分析和处理了。下面是一些常见的数据处理操作示例：

计算平均成绩

要计算学生的平均成绩，我们可以遍历每一行数据，并将每个学生的各科成绩累加起来，然后除以科目数。以下是计算平均成绩的代码示例：

def calculate_average_score(rows):
    num_subjects = len(rows[0]) - 2  # 减去姓名和学号列
    total_scores = [0] * num_subjects
    for row in rows[1:]:  # 跳过表头
        for i, score in enumerate(row[2:]):  # 从第3列开始计算
            total_scores[i] += int(score)
    average_scores = [round(total_score / len(rows), 2) for total_score in total_scores]
    return average_scores

average_scores = calculate_average_score(rows)
print(average_scores)

运行以上代码，输出结果为：