C++实现保存数据至EXCEL

目录
  • 场景需求
  • 功能函数代码
  • C++测试代码
  • 测试效果

场景需求

在用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();
}

测试效果

图1 随机生成的图像矩阵

图2 保存效果图

如上图所示,为了方便,我随机生成了一个1000*1000的图像矩阵,并定义了第二行第三列的点为nan值,保存效果如图2所示,成功!

如果函数有什么可以改进完善的地方,非常欢迎大家指出~

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

(0)

相关推荐

随机推荐