C++中如何将数据保存为CSV文件

时间:2022-11-12 Mz枫人气:0

C++将数据保存为CSV文件

因为最近涉及到保存模型推理结果的输出文件，所以学一学如何将数据保存为CSV文件，比如保存检测框box的顶点，置信度，label，入侵检测的结果等。

用到的也是C++的ofstream，ofstream有一个操作是"<<"，这个也很好用，就类似std::cout的操作一样即可。

比如我现在的数据是比较统一的，每一个样本是一行，一行数据要分成四列，第一列是样本的图像地址，第二列是标签，第三列是最终预测值，第四列是概率，每一列的格式是一样的

那么我的代码就是这样：

ofstream file(CSV_PATH);
if (file)
{
    file << image_path << "," << label << "," << prediction << "," << probability << "\n";
}
file.close();

要注意的是，逗号表示的是换列，换行符号就是换行。

CSV文件可以用excel直接打开

如何存储CSV文件

应用工程里，经常会遇到存储一些数据，存储下来进行分析

#include <iostream>
#include <fstream>

using namespace std;

std::string CSV_PATH = "./data.csv";
ofstream csv;
int frame_num = 0;

struct Name
{
    int age;
    float height;
    float weight;
    int score;
};

int main()
{
    Name Zhangsan{30, 1.75, 78, 98};
    csv.open(CSV_PATH);

    while (true)
    {
        if (csv.is_open())
        {
            frame_num++;
            if (frame_num == 1)
            {
                csv << "age"
                    << ","
                    << "height(m)"
                    << ","
                    << "weight(kg)"
                    << ","
                    << "score"
                    << "\n";
            }
            else
            {
                csv << Zhangsan.age << "," << Zhangsan.height << "," << Zhangsan.weight << "," << Zhangsan.score
                    << "\n";
            }
        }
        if (frame_num > 10000) {
            csv.close();
        }
    }

    return 0;
}

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。

加载全部内容