场景需求
在用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所示，成功！
如果函数有什么可以改进完善的地方，非常欢迎大家指出，一同进步何乐而不为呢~
如果文章帮助到你了，可以点个赞让我知道，我会很快乐~加油！
