python csv添加两列单元格

原创

mob64ca12dea1dc 2024-07-31 08:57:37 ©著作权

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

使用Python CSV添加两列单元格的完整指南

CSV（Comma-Separated Values）是一种常见的数据存储格式，由于其简单易读，人们广泛使用Python进行操作。在许多项目中，可能会遇到需要向现有CSV文件中添加新列的情况。本文将详细介绍如何使用Python CSV模块来实现这一功能，并提供代码示例。

引言

在数据处理的过程中，添加新列是一个相对简单但却常见的操作。无论是增加新的数据特征还是进行数据校对，这一功能都是不可或缺的。下面，我们将阐述如何实现这一目标。

代码实现

首先，我们需要导入Python的CSV模块，并准备一个简单的CSV文件。假设我们的初始CSV文件中包含一些学生的姓名和成绩。示例数据如下：

Name,Score
Alice,85
Bob,90
Charlie,87

我们希望在此基础上添加两列：Grade（学年）和Pass/Fail（合格/不合格）。让我们先定义这两个新列的计算规则：

Grade: 当分数大于或等于90时为“A”，80到89之间为“B”，70到79之间为“C”，否则为“D”。
Pass/Fail: 分数大于或等于60时为“Pass”，否则为“Fail”。

接下来，是完整的代码示例：

import csv

# 读取原始CSV文件
with open('students.csv', mode='r') as infile:
    reader = csv.reader(infile)
    # 获取表头
    header = next(reader)
    # 添加新的列名
    header.extend(['Grade', 'Pass/Fail'])

    # 处理数据并添加新的行
    rows = []
    for row in reader:
        name, score = row
        score = int(score)  # 将字符串转换为整数

        # 根据分数计算Grade及Pass/Fail
        if score >= 90:
            grade = 'A'
        elif score >= 80:
            grade = 'B'
        elif score >= 70:
            grade = 'C'
        else:
            grade = 'D'

        pass_fail = 'Pass' if score >= 60 else 'Fail'

        # 将新的数据行添加到列表中
        rows.append([name, score, grade, pass_fail])

# 将新的数据写入新的CSV文件
with open('students_with_grades.csv', mode='w', newline='') as outfile:
    writer = csv.writer(outfile)
    # 写入表头
    writer.writerow(header)
    # 写入数据行
    writer.writerows(rows)

代码讲解

在上面的代码中，我们首先打开并读取原始的CSV文件。通过csv.reader读取，使用next(reader)获取CSV的表头。接着，我们为表头添加了新的列名。

接下来，在循环中，我们逐行读取数据，计算每个学生的Grade和Pass/Fail。根据分数的不同，我们为每位学生生成相应的类别，并将结果添加到一个列表中。

最后，我们创建一个新的CSV文件，将更新后的数据写入其中，包括新添加的列。

状态图

以下是数据处理过程的状态图，展示了从读取CSV文件到写出新文件的状态变化：

stateDiagram
    [*] --> 读取文件
    读取文件 --> 获取表头
    获取表头 --> 添加新列名
    添加新列名 --> 逐行处理数据
    逐行处理数据 --> 计算Grade和Pass/Fail
    计算Grade和Pass/Fail --> 添加到行数据列表
    添加到行数据列表 --> 写入新CSV文件
    写入新CSV文件 --> [*]