学无先后,达者为师

网站首页 编程语言 正文

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

作者:Mz枫   更新时间: 2022-12-10 编程语言

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;
}

原文链接:https://blog.csdn.net/u012456019/article/details/101481424

栏目分类
最近更新