将特征矩阵转换为三元组形式C++ [英] Convert an Eigen matrix to Triplet form C++

问题描述

我认为Eigen使用压缩方法存储稀疏矩阵。有没有什么方法可以从STD：：Vectors中提取特征稀疏矩阵的三元组格式的向量？

谢谢。

详细信息(三元组格式示例) 矩阵的三元组格式：

A=
3 0 4 0
0 0 1 0
0 2 0 5
4 0 0 0

i = 1 1 2 3 3 4  // row
j = 1 3 3 2 4 1 // column
S = 3 4 1 2 5 4  // values

推荐答案

简单如tutorial：

#include <Eigen/Sparse>
#include <iostream>

using namespace Eigen;
using std::cout;
using std::endl;

typedef Triplet<int> Trip;

int main(int argc, char *argv[]){

    std::vector<Trip> trp, tmp;

    // I subtracted 1 from the indices so that the output matches your question
    trp.push_back(Trip(1-1,1-1,3));
    trp.push_back(Trip(1-1,3-1,4));
    trp.push_back(Trip(2-1,3-1,1));
    trp.push_back(Trip(3-1,2-1,2));
    trp.push_back(Trip(3-1,4-1,5));
    trp.push_back(Trip(4-1,1-1,4));

    int rows, cols;
    rows = cols = 4;
    SparseMatrix<int> A(rows,cols);

    A.setFromTriplets(trp.begin(), trp.end());
    cout << "Matrix from triplets:" << endl;
    cout << A << endl;            

    cout << endl << "Triplets:" << endl;
    cout << "Row	Col	Val" <<endl;
    for (int k=0; k < A.outerSize(); ++k)
    {
        for (SparseMatrix<int>::InnerIterator it(A,k); it; ++it)
        {
            cout << 1+it.row() << "	"; // row index
            cout << 1+it.col() << "	"; // col index (here it is equal to k)
            cout << it.value() << endl;
        }
    }

    return 0;
}

这篇关于将特征矩阵转换为三元组形式C++的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！