欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

[Eigen]Eigen将矩阵写至文件与读文件的简单例子

程序员文章站 2022-03-02 12:40:31
...

代码

#include <iostream>
#include <vector>
#include <Eigen/Dense>
#include <fstream>

void ReadData(std::istream &fin, Eigen::MatrixXd &m_matrix)
{
	int numRow = m_matrix.rows();
	int numCol = m_matrix.cols();

	Eigen::VectorXd vecPerRow(numRow);
	for (int j = 0; j < numRow; j++)//共numRow行
	{
		for (int i = 0; i < numCol; i++)//共numCol列组成一行
		{
			fin >> m_matrix(j, i);
		}
		
	}	
}
//blog.csdn.net/xinshuwei/article/details/94064790

int main()
{
	Eigen::MatrixXd m_matrix, m_matrixRead;
	m_matrix.resize(4, 3);
	m_matrix.fill(0);
	m_matrixRead.resize(4, 3);
	std::cout << m_matrix << std::endl;

	/*写文件**/
	std::ofstream fout("matrixTest.bin", std::ios::binary);
	fout << m_matrix << std::endl;
	fout.flush();
	//Eigen::VectorXd vecPerRow(4);
	//std::cout << vecPerRow << std::endl;

	/*读文件**/
	std::ifstream fin("matrixTest.bin", std::ios::binary);
	if (!fin)
	{
		return 0;
	}
	
	ReadData(fin, m_matrixRead);
	std::cout <<"Matrix read from file:\n"<< m_matrixRead << std::endl;
	
	return 0;
}

要点

  • Eigen重载了操作符"<<",可以在iostream中直接打印矩阵
  • 也可以通过重载操作符">>"达到iostream读文件中的矩阵进入
  • 写的时候好像是按列写入的,读好像是按行读取的,具体机制没有深究
相关标签: Matlab