C++标准模板库实现CSV文件读写

职业规划徐老师

802人浏览 · 2025-08-21 12:04:55

职业规划徐老师 · 2025-08-21 12:04:55 发布

简介：CSV文件作为数据交换的通用格式，在IT领域应用广泛。本文将详细展示如何使用C++的STL功能，尤其是 vector 容器，来高效读取和写入CSV文件。文章涵盖了CSV基础概念、C++和STL的介绍、文件读写的具体实现，以及对CSV文件的常用操作如创建、插入、修改、删除数据，以及求和和平均值计算。此外，还讨论了在实现过程中需要特别注意的边界条件处理。
C++读写CSV文件

1. CSV文件基础概念

CSV（Comma-Separated Values，逗号分隔值）文件是一种纯文本文件，用于存储表格数据，其数据项通常以逗号分隔。每行代表一个数据记录，每个记录中的数据项按照字段顺序排列，字段之间用逗号分隔。CSV文件是数据交换的一种简单格式，因其跨平台兼容性强、易于创建和解析而广泛应用于数据导出导入和数据备份等领域。由于CSV文件的简洁性，几乎所有的编程语言都可以轻松地对CSV文件进行读写操作。

示例CSV内容：
姓名,年龄,职业
张三,28,工程师
李四,30,设计师
王五,25,产品经理

在进行进一步的C++操作之前，了解CSV文件结构对确保数据正确读取和写入至关重要。本章主要介绍了CSV文件的基本知识，为后续章节中使用C++处理CSV文件奠定了基础。

2. C++与STL概述

2.1 C++标准模板库（STL）简介

2.1.1 STL的核心组成

C++标准模板库（STL）是C++语言的核心部分，它提供了一套泛型（template-based）的容器（containers）、迭代器（iterators）、算法（algorithms）和函数对象（function objects）等组件。通过这些组件，C++的STL实现了数据结构和操作的抽象化，允许程序员在编写代码时不必从头开始构造数据结构，而是可以直接利用STL提供的现成组件来处理数据集合，提高开发效率。

STL的核心在于三个主要部分：

容器：用于存储数据的泛型数据结构，如 vector 、 list 、 map 、 set 等。它们可以容纳不同类型的对象，并提供丰富的接口来进行元素的增删改查等操作。
迭代器 ：作为容器和算法之间的桥梁，迭代器提供了访问容器内元素的方式，并可以遍历容器。
算法：一组用于处理容器中元素的模板函数，它们是高度通用的，能够在不同类型的容器上执行相同的操作，例如排序、查找、计数、复制等。

2.1.2 STL中的容器、迭代器和算法

STL中的容器、迭代器和算法紧密集成在一起，为处理数据集合提供了强大的工具。下面我们将分别详细介绍这三者。

容器

STL容器可以分为两大类：顺序容器和关联容器。顺序容器如 vector 、 deque 、 list 等，能够保持元素的插入顺序，并提供快速的随机访问。关联容器如 map 、 set 、 multimap 、 multiset 等，基于键值对，并保持数据排序，适合于快速查找。

迭代器

迭代器的种类有很多，包括输入迭代器、输出迭代器、前向迭代器、双向迭代器和随机访问迭代器。它们分别对应不同的操作能力和复杂度。迭代器的主要作用是使得算法不必关心容器的内部数据结构，可以将算法与容器解耦。

算法

STL中的算法是泛型的，这意味着它们可以应用于任何STL容器，并且能够处理几乎任何类型的数据。算法通过迭代器对容器内的数据进行操作，常见的算法包括 sort 、 find 、 count 、 transform 、 copy 等。

2.2 C++的基础语法回顾

2.2.1 变量和数据类型

在C++中，变量是用于存储数据的标识符，它们必须在使用前声明，并具有特定的数据类型。C++的数据类型分为基本数据类型和复合数据类型。基本数据类型包括整型、浮点型、字符型和布尔型等，复合数据类型则包括数组、结构体、联合体、枚举以及C++特有的类类型。

声明变量的语法如下：

type identifier = value;

其中， type 是数据类型， identifier 是变量名， value 是可选的初始化值。

控制结构和函数

控制结构用于决定程序执行的流程，包括条件语句（如 if 、 switch ）和循环语句（如 while 、 for ）。函数是组织代码、重用代码的基本单位，它们将执行特定任务的代码块封装起来，并可以接受输入参数和返回结果。

函数的基本语法结构如下：

type function_name(parameters) {
    // Function body
}

其中， type 是返回值类型， function_name 是函数名， parameters 是参数列表， {} 内是函数体。

C++ 还支持函数重载，即同一个作用域中可以声明同名函数，但函数参数类型、个数或顺序必须不同，从而根据不同的调用情况执行不同的功能。此外，C++11引入了lambda表达式，允许定义匿名函数对象，这为编程提供了更大的灵活性。

3. 使用C++读取CSV文件的方法

在处理数据时，读取CSV文件是一种常见的需求。CSV（Comma-Separated Values，逗号分隔值）格式因其简洁性和易读性而广泛应用于数据交换。C++是一种高性能的编程语言，它没有内置的CSV文件解析库，但我们可以利用C++强大的标准模板库（STL）以及字符串处理功能来实现高效且灵活的CSV文件读取。

3.1 手动解析CSV文件

手动解析CSV文件意味着我们需要自己编写代码来处理字符串分割、转义字符等问题。这种方法虽然复杂，但提供了最大的灵活性。

3.1.1 分割字符串的基本原理

在CSV文件中，数据项通常由逗号分隔。为了将每一行的数据分割成单独的字段，我们可以采用一种基于迭代器的方法。通过迭代器遍历字符串，并识别逗号、换行符等特殊字符，我们可以实现字符串的分割。

#include <iostream>
#include <string>
#include <vector>

std::vector<std::string> split(const std::string& s, char delimiter) {
    std::vector<std::string> tokens;
    std::string token;
    std::istringstream tokenStream(s);

    while (std::getline(tokenStream, token, delimiter)) {
        tokens.push_back(token);
    }

    return tokens;
}

int main() {
    std::string csv_line = "aaa,bbb,ccc\n";
    std::vector<std::string> fields = split(csv_line, ',');

    for (const auto& field : fields) {
        std::cout << field << std::endl;
    }

    return 0;
}

上述代码中，我们定义了一个 split 函数，它接受一个字符串和一个分隔符，并返回一个包含分割后各字段的字符串向量。这个函数使用了 std::getline 来逐个读取被分隔符分隔的字段，并将其存储到向量中。

3.1.2 字符串流（iostream）的应用

C++中的字符串流（ std::istringstream ）可以用来读取和解析字符串。当我们将CSV内容读入字符串后，就可以利用字符串流的特性来解析数据。

#include <sstream>
#include <iostream>

int main() {
    std::string csv_line = "aaa,bbb,ccc\n";
    std::istringstream stream(csv_line);

    std::string field;
    while (getline(stream, field, ',')) {
        std::cout << field << std::endl;
    }

    return 0;
}

在这个例子中，我们首先创建了一个 istringstream 对象，并将CSV内容作为字符串传递给它。然后，我们使用 getline 函数迭代地读取每个字段，直到字符串结束。

3.2 利用STL简化CSV读取过程

利用STL中的容器和迭代器可以大幅简化读取CSV文件的过程。STL中的 std::vector 和 std::map 容器可以用来存储数据，并且提供方便的接口来访问和操作这些数据。

3.2.1 使用vector存储CSV数据

使用 std::vector 存储CSV数据是一种非常直观的方法。对于每一行数据，我们可以将其作为一个 std::vector<std::string> 对象读取，整个CSV文件则可以用一个 std::vector<std::vector<std::string>> 来表示。

#include <iostream>
#include <fstream>
#include <vector>
#include <sstream>

std::vector<std::vector<std::string>> read_csv_to_vector(const std::string& file_path) {
    std::vector<std::vector<std::string>> result;
    std::ifstream file(file_path);

    std::string line;
    while (getline(file, line)) {
        result.push_back(split(line, ','));
    }

    return result;
}

int main() {
    std::string file_path = "data.csv";
    auto csv_data = read_csv_to_vector(file_path);

    // 打印CSV数据
    for (const auto& row : csv_data) {
        for (const auto& field : row) {
            std::cout << field << " ";
        }
        std::cout << std::endl;
    }

    return 0;
}

3.2.2 使用map关联数据项

如果CSV文件中包含标题行，我们可以使用 std::map 来存储数据，这样可以通过标题直接访问对应的数据项。

#include <iostream>
#include <fstream>
#include <map>
#include <sstream>
#include <string>

std::map<std::string, std::string> read_csv_line_to_map(const std::string& line) {
    std::map<std::string, std::string> result;
    std::istringstream line_stream(line);
    std::string cell;

    getline(line_stream, cell, ',');
    size_t pos = 0;

    while (getline(line_stream, cell, ',')) {
        result[cell.substr(0, pos)] = cell.substr(pos + 1);
        pos = line.find(',', pos) + 1;
    }

    return result;
}

int main() {
    std::string file_path = "data.csv";
    std::ifstream file(file_path);
    std::string line;

    if (getline(file, line)) {
        auto header_map = read_csv_line_to_map(line);
        while (getline(file, line)) {
            auto data_map = read_csv_line_to_map(line);
            for (const auto& header : header_map) {
                std::cout << header.first << ": " << data_map[header.first] << std::endl;
            }
            std::cout << std::endl;
        }
    }

    return 0;
}

在这个示例中， read_csv_line_to_map 函数首先读取标题行，并将其存储在一个map中。对于文件中的每一行数据，我们调用此函数，并使用标题行中的标题来访问数据项。

通过上述章节，我们已经了解了如何在C++中手动解析CSV文件以及如何利用STL简化这一过程。下一章将介绍如何使用C++向CSV文件写入数据。

4. 使用C++写入CSV文件的方法

4.1 CSV文件格式化写入

CSV（Comma-Separated Values，逗号分隔值）文件是一种常见的文本文件格式，用于存储表格数据，例如数字和文本。CSV文件通常由一系列的记录组成，每条记录是数据项的列表，各数据项之间用逗号分隔。一个完整的CSV文件可能包含若干行，每行代表一条记录，记录中的每个字段则通过特定的分隔符（通常是逗号）进行分隔。

4.1.1 构造符合CSV规范的字符串

在C++中构造符合CSV规范的字符串，需要特别注意以下几个方面：

特殊字符处理 ：如果字段中包含逗号（,）、换行符（\n）、引号（”）或制表符（\t）等特殊字符，它们需要被适当地转义或者引用，以避免解析错误。例如，如果一个字段内包含逗号，则整个字段需要用双引号（”）包裹。
引号转义 ：如果字段内包含双引号，则需要使用两个双引号（”“）进行转义。
换行处理 ：需要考虑不同操作系统间换行符的兼容性。通常情况下，Windows系统使用CRLF（\r\n）作为换行符，而Unix/Linux系统使用LF（\n）作为换行符。

以下是一个简单的例子，展示了如何手动格式化字符串为CSV格式：

#include <iostream>
#include <string>

std::string EscapeCSVField(const std::string &field) {
    // 如果字段包含特殊字符，需要转义或引用
    std::string escaped;
    escaped.push_back('"'); // 开始引用

    for (char ch : field) {
        if (ch == ',' || ch == '"' || ch == '\n') {
            escaped.push_back('"'); // 转义特殊字符
        }
        escaped.push_back(ch);
    }

    escaped.push_back('"'); // 结束引用
    return escaped;
}

std::string CreateCSVString(const std::vector<std::string> &data) {
    std::string csvLine;
    for (size_t i = 0; i < data.size(); ++i) {
        if (i > 0) csvLine += ", ";
        csvLine += EscapeCSVField(data[i]);
    }
    return csvLine;
}

int main() {
    std::vector<std::string> lineData = {"This", "is a test", "of, comma, separation"};
    std::string csvLine = CreateCSVString(lineData);
    std::cout << csvLine << std::endl;
    return 0;
}

4.1.2 文件流（fstream）操作

当需要写入CSV文件时，我们会用到C++的文件流库 fstream 。使用 fstream ，我们可以打开一个文件，向其中写入数据，然后关闭它。下面是一个创建和写入CSV文件的例子：

#include <fstream>
#include <iostream>

int main() {
    std::string filename = "output.csv";
    std::ofstream file(filename); // 创建并打开文件用于写入

    if (file.is_open()) {
        std::vector<std::string> data = {"John", "Doe", "456 Elm Street", "Anytown", "12345"};
        for (const std::string &str : data) {
            file << str << ','; // 写入数据项
        }
        file.seekp(-1, std::ios_base::end); // 移动文件指针到末尾前一个字符
        file.put('\n'); // 写入换行符
    } else {
        std::cerr << "Unable to open file for writing." << std::endl;
    }

    file.close(); // 关闭文件
    return 0;
}

需要注意的是，在实际的CSV写入操作中，还需要处理字段的转义和引用规则，以及多条记录的管理，保持代码的健壮性和可读性。

4.2 使用STL容器批量写入

在处理大量数据时，使用STL容器可以简化操作并提高效率。 std::vector 和 std::map 等容器可以方便地存储和管理数据，然后通过循环遍历它们来写入CSV文件。

4.2.1 利用vector输出数据

std::vector 是一个动态数组，能够存储任意类型的数据。当我们有多个数据项需要写入CSV文件时，可以使用 std::vector 来管理这些数据项。

#include <fstream>
#include <vector>
#include <string>

int main() {
    std::string filename = "output.csv";
    std::ofstream file(filename); // 打开文件进行写入

    if (file.is_open()) {
        std::vector<std::vector<std::string>> records = {
            {"ID", "Name", "Address"},
            {"001", "John Doe", "123 Main St"},
            {"002", "Jane Smith", "456 Maple Ave"}
        };
        for (const auto &record : records) {
            for (const auto &field : record) {
                file << field << ',';
            }
            file.seekp(-1, std::ios_base::end); // 移动文件指针到末尾前一个字符
            file.put('\n'); // 写入换行符
        }
    } else {
        std::cerr << "Unable to open file for writing." << std::endl;
    }

    file.close(); // 关闭文件
    return 0;
}

4.2.2 使用map输出键值对数据

std::map 是一个关联容器，它存储的元素是一个键值对，其中的键是唯一的。在某些情况下，数据可能更适合以键值对的形式组织，这样可以很轻松地通过键来访问对应的值。

#include <fstream>
#include <map>
#include <string>

int main() {
    std::string filename = "output.csv";
    std::ofstream file(filename); // 打开文件进行写入

    if (file.is_open()) {
        std::map<std::string, std::string> record = {
            {"Name", "John Doe"},
            {"Address", "123 Main St"},
            {"Phone", "555-1234"}
        };
        for (const auto &kv : record) {
            file << kv.first << ": " << kv.second << ',';
        }
        file.seekp(-1, std::ios_base::end); // 移动文件指针到末尾前一个字符
        file.put('\n'); // 写入换行符
    } else {
        std::cerr << "Unable to open file for writing." << std::endl;
    }

    file.close(); // 关闭文件
    return 0;
}

在使用STL容器进行CSV写入时，文件流（fstream）对象负责打开和关闭文件，而数据则存储在STL容器中。通过遍历容器中的元素，我们可以把它们格式化为CSV格式并写入文件中。在实践中，我们可能还会根据具体需求进行相应的错误处理和异常管理，确保数据正确无误地输出。

5. 支持的CSV操作功能

5.1 错误处理和异常管理

在处理CSV文件时，错误处理和异常管理是十分重要的环节。它们保证了程序的健壮性和可靠性，特别是在数据量大和来源多样化的情况下。

5.1.1 捕获和处理I/O异常

在C++中，使用 try-catch 块来捕获异常。当进行文件I/O操作时，如文件不存在、权限问题或其他I/O错误，可以通过捕获异常来进行相应的处理。

下面的代码演示了如何捕获和处理在尝试打开文件时可能出现的异常：

#include <iostream>
#include <fstream>
#include <string>

int main() {
    std::string filename = "data.csv";
    std::ifstream file(filename);

    try {
        if (!file.is_open()) {
            throw std::runtime_error("Could not open file: " + filename);
        }

        // 文件读取操作...

        file.close();
    } catch (const std::exception& e) {
        std::cerr << "Exception: " << e.what() << '\n';
    }

    return 0;
}

5.1.2 CSV数据格式校验

CSV数据格式校验是指检查CSV文件是否符合标准的格式规范，例如字段是否被正确的引号包围、字段内是否包含分隔符等。下面的示例代码提供了简单的校验功能：

#include <fstream>
#include <sstream>
#include <string>

bool isCSVValid(const std::string& line) {
    std::istringstream s(line);
    std::string field;
    bool quoted = false;
    while (std::getline(s, field, ',')) {
        if (field.size() > 1 && field.front() == field.back() && field.front() == '"') {
            field = field.substr(1, field.size() - 2);
            quoted = !quoted;
        }
        if (field.find(',') != std::string::npos && !quoted) {
            return false; // 未被引号包围的逗号表示格式错误
        }
    }
    return true;
}

5.2 CSV文件的高级操作

5.2.1 文件合并与分割

文件合并是指将多个CSV文件合并为一个，而文件分割是指将一个大CSV文件分割为多个更小的部分。

文件合并

合并CSV文件通常涉及读取多个文件的内容并顺序写入到一个新的文件中。这里是一个简单的合并两个CSV文件的例子：

#include <fstream>
#include <vector>
#include <string>

int main() {
    std::ifstream file1("file1.csv");
    std::ifstream file2("file2.csv");
    std::ofstream outfile("merged.csv");

    std::string line;
    // 读取第一个文件内容
    while (getline(file1, line)) {
        outfile << line << "\n";
    }
    // 读取第二个文件内容
    while (getline(file2, line)) {
        outfile << line << "\n";
    }

    file1.close();
    file2.close();
    outfile.close();

    return 0;
}

文件分割

分割CSV文件一般根据行数或大小来执行。下面的代码展示了如何根据行数来分割一个CSV文件：

#include <iostream>
#include <fstream>
#include <string>

void splitCSV(const std::string& inputFilename, const std::string& outputFilename, size_t rows) {
    std::ifstream file(inputFilename);
    std::ofstream outfile(outputFilename);
    std::string line;
    size_t count = 0;

    while (getline(file, line)) {
        outfile << line << "\n";
        ++count;
        if (count >= rows) {
            count = 0;
            outfile.close();
            // 为下一片段创建新文件
            outfile.open(std::to_string(rows * splitCount++) + outputFilename);
        }
    }
    file.close();
}

int main() {
    std::string inputFilename = "bigfile.csv";
    std::string outputFilename = "chunk-";
    size_t rowsPerFile = 1000; // 每个文件1000行

    splitCSV(inputFilename, outputFilename, rowsPerFile);

    return 0;
}

5.2.2 数据排序和筛选

数据排序和筛选是处理CSV文件时常见的需求。可以利用C++标准库中的算法对数据进行排序，使用标准库容器或自定义结构来筛选数据。

数据排序

下面的代码展示了如何使用 std::sort 对CSV文件中的数据行进行排序：

#include <fstream>
#include <sstream>
#include <vector>
#include <algorithm>

// 用于表示CSV文件中的一行
struct CSVRow {
    std::vector<std::string> data;

    // 重载<运算符以便于排序
    bool operator<(const CSVRow& that) const {
        return data < that.data;
    }
};

int main() {
    std::ifstream file("data.csv");
    std::vector<CSVRow> rows;

    std::string line;
    // 读取CSV文件的所有行
    while (std::getline(file, line)) {
        std::istringstream s(line);
        std::string field;
        CSVRow row;
        while (std::getline(s, field, ',')) {
            row.data.push_back(field);
        }
        rows.push_back(row);
    }

    // 对行进行排序
    std::sort(rows.begin(), rows.end());

    // 输出排序后的数据
    for (const auto& row : rows) {
        for (const auto& field : row.data) {
            std::cout << field << ',';
        }
        std::cout << '\n';
    }

    return 0;
}

数据筛选

筛选数据则需要定义筛选条件，然后应用这些条件到数据集上，以下示例展示了如何筛选出特定列值大于某个阈值的行：

#include <fstream>
#include <vector>
#include <sstream>
#include <iostream>
#include <algorithm>

// 用于表示CSV文件中的一行
struct CSVRow {
    std::vector<std::string> data;

    // 比较操作，用于筛选
    bool checkCondition(int columnIndex, std::string conditionValue) {
        return data[columnIndex] > conditionValue;
    }
};

int main() {
    std::ifstream file("data.csv");
    std::vector<CSVRow> rows;

    std::string line;
    // 读取CSV文件的所有行
    while (std::getline(file, line)) {
        std::istringstream s(line);
        std::string field;
        CSVRow row;
        while (std::getline(s, field, ',')) {
            row.data.push_back(field);
        }
        rows.push_back(row);
    }

    // 筛选数据
    std::string conditionValue = "50"; // 假设筛选条件是某列的值大于50
    std::vector<CSVRow> filteredRows;
    for (const auto& row : rows) {
        if (row.checkCondition(0, conditionValue)) { // 假设0是需要筛选的列的索引
            filteredRows.push_back(row);
        }
    }

    // 输出筛选后的数据
    for (const auto& row : filteredRows) {
        for (const auto& field : row.data) {
            std::cout << field << ',';
        }
        std::cout << '\n';
    }

    return 0;
}

请注意，上述代码仅用作示例，实际应用中应根据具体的需求进行调整。这些高级操作展示了如何在C++中处理CSV文件，从而能够实现复杂的数据操作任务。

本文还有配套的精品资源，点击获取