python 一列csv数据相加

原创

mob64ca12f0cf8f 2023-12-28 08:53:20 ©著作权

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

Python实现CSV数据相加

前言

CSV（Comma-Separated Values）是一种常用的数据格式，它以逗号作为字段分隔符，以换行符作为记录分隔符，可以用来存储、传输和处理表格数据。在实际数据处理中，我们经常需要对CSV文件进行操作，比如合并多个CSV文件的数据。本文将介绍如何使用Python来实现一列CSV数据的相加操作。

CSV文件的读取与写入

首先，我们需要了解如何使用Python读取和写入CSV文件。Python的csv模块提供了简单而强大的功能来处理CSV文件。我们可以使用csv.reader来读取CSV文件，并使用csv.writer来写入CSV文件。

以下是一个示例代码，用于读取名为"input.csv"的CSV文件，并将数据保存到一个列表中：

import csv

data = []

with open('input.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        data.append(row)

请注意，我们使用了with语句来自动关闭文件，这样可以确保资源的释放和文件的正确关闭。

接下来，我们可以使用csv.writer将数据写入CSV文件。以下是一个示例代码，将数据写入名为"output.csv"的CSV文件：

import csv

data = [
    ['A', 'B', 'C'],
    [1, 2, 3],
    [4, 5, 6]
]

with open('output.csv', 'w') as file:
    csv_writer = csv.writer(file)
    for row in data:
        csv_writer.writerow(row)

一列CSV数据的相加

在实际场景中，我们经常遇到需要对一列CSV数据进行相加的情况。例如，我们有一列数据，每一行都是一个数字，我们希望将这些数字相加得到一个总和。以下是一个示例代码，演示了如何实现这个功能：

import csv

total = 0

with open('input.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        total += int(row[0])

print(f'The sum of the column is: {total}')

在上述代码中，我们首先初始化一个变量total，用于保存相加的结果。然后，我们使用csv.reader读取CSV文件，并遍历每一行数据。对于每一行数据，我们从中取出第一个元素（即当前行的数字），并将其转换为整数类型。最后，我们将这个数字加到total中，并最终输出结果。

序列图

为了更清晰地说明代码的执行过程，我们可以使用序列图来展示。

以下是一个使用mermaid语法绘制的序列图，展示了读取CSV文件和相加操作的过程：

sequenceDiagram
    participant Python
    participant CSV File
    
    Python->>CSV File: 打开文件
    CSV File->>Python: 返回文件对象
    Python->>CSV File: 读取数据
    CSV File->>Python: 返回数据列表
    Python->>Python: 初始化总和变量
    loop 遍历数据
        Python->>Python: 相加操作
    end
    Python->>Python: 输出结果