本征稀疏矩阵。相对于行对非零元素进行排序

Question

我有一个稀疏矩阵，我需要按行的升序存储非零元素和相应的行、列

关注这个 https://eigen.tuxfamily.org/dox/group__TutorialSparse.html 我试过了

// mat =
// 1.0, 0.0, 1.0, 0.0, 0.0
// 1.0, 0.0, 0.0, 0.0, 1.0
// 0.0, 1.0, 0.0, 0.0, 0.0
// 0.0, 0.0, 1.0, 0.0, 1.0
// 1.0, 0.0, 0.0, 1.0, 0.0 

// create and fill the sparse matrix
SparseMatrix<double> mat(5,5);
mat.insert(0,0) = 1.0;
mat.insert(0,2) = 1.0;
mat.insert(1,0) = 1.0;
mat.insert(1,4) = 1.0;
mat.insert(2,1) = 1.0;
mat.insert(3,2) = 1.0;
mat.insert(3,4) = 1.0;
mat.insert(4,0) = 1.0;
mat.insert(4,3) = 1.0;

//matrix where to store the row,col,value
Eigen::MatrixXd mat_map(mat.nonZeros(),3);

int index_mat;
index_mat=-1;

for (int k=0; k<mat.outerSize(); ++k)
  for (SparseMatrix<double>::InnerIterator it(mat,k); it; ++it)
   {
     index_mat++;
     mat_map(index_mat,0) = it.row();   // row index
     mat_map(index_mat,1) = it.col();   // col index 
     mat_map(index_mat,2) = it.value();
   }


cout << mat_map << endl;

我得到的是下面的

而我想要的是

任何帮助将不胜感激

谢谢！

Answer 1

SparseMatrix 有一个可选的模板参数，用于定义存储顺序（参见 eigen documentation）。默认情况下，SparseMatrix 使用 column-major。因此，不要使用 Eigen::SparseMatrix<double>，而是使用 Eigen::SparseMatrix<double, Eigen::RowMajor>。

最好为此使用别名

using MyMatrix = Eigen::SparseMatrix<double, Eigen::RowMajor>;
MyMatrix mat(5,5);

本征稀疏矩阵。相对于行对非零元素进行排序

Eigen Sparse matrix. Sorting non-zero elements with respect to rows

c++

sparse-matrix

eigen