C++实现保存数据至EXCEL

翟天保Steven 2022-11-12 我要评论

场景需求

在用C++编程时，往往会有保存数据的需求，比如我做图像处理的工作，经常会想将图片的像素数据保存至Excel中，然后用matlab进行分析或者用于其他方面。

为了解决该需求，本文将为大家提供一个保存数据至EXCEL的函数。我以保存一幅图像数据为例，大家可以根据自己需求进行微调。

功能函数代码

void WritePicToExcel(string name, cv::Mat pic)
{
    // 我这里写了断言，可以根据需要自己调整
	CV_Assert(pic.type() == CV_32FC1);
	ofstream outfile(name);
	int row = pic.rows;
	int col = pic.cols;
	for (int i = 0; i < row; i++)
	{
        // 采用指针读取Mat矩阵的方式提速
		float *p = pic.ptr<float>(i);
		for (int j = 0; j < col; j++)
		{
            // 若需要保存的数据是非nan值，则正常保存；若为nan值，则保存字符“nan”
			if (p[j] == p[j])
			{
				outfile << p[j] << (j == (col - 1) ? '\n' : '\t');
			}
			else {
				outfile << "nan" << (j == (col - 1) ? '\n' : '\t');
			}
			
		}
	}
	outfile.close();
}

C++测试代码

#include<iostream>
#include<fstream>
#include<opencv2/opencv.hpp>
using namespace std;
void WritePicToExcel(string name, cv::Mat pic);
int main()
{
	cv::Mat pic = cv::Mat::zeros(1000, 1000, CV_32FC1);
	for (int i = 0; i < pic.rows; i++)
	{
		for (int j = 0; j < pic.cols; j++)
		{
			pic.at<float>(i, j) = rand() % 255;
		}
	}
    // 人为设置一个nan值检验是否可以有效保存
	pic.at<float>(1, 2) = nan("");
    // 输入的string为保存的文件名，函数其实还可以加点文件格式的判断，有兴趣的自己改进下吧
	WritePicToExcel("test1.xls", pic);
	return 0;
}
void WritePicToExcel(string name, cv::Mat pic)
{
	CV_Assert(pic.type() == CV_32FC1);
	ofstream outfile(name);
	int row = pic.rows;
	int col = pic.cols;
	for (int i = 0; i < row; i++)
	{
		float *p = pic.ptr<float>(i);
		for (int j = 0; j < col; j++)
		{
			if (p[j] == p[j])
			{
				outfile << p[j] << (j == (col - 1) ? '\n' : '\t');
			}
			else {
				outfile << "nan" << (j == (col - 1) ? '\n' : '\t');
			}
			
		}
	}
	outfile.close();
}