将特征矩阵转换为C++形式的三元组

Convert an Eigen matrix to Triplet form C++

本文关键字：三元组 C++ 特征 转换      更新时间：2023-10-16

我认为Eigen使用压缩方法来存储稀疏矩阵。有什么方法可以从std:：vectors中提取特征稀疏矩阵的Triplet格式向量吗？

谢谢。

更多信息（三元组格式示例）矩阵的Triplet格式：

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

问题的答案是：

// Is there some method such as:
std::vector<Eigen::Triplet<double>> T = SparseMat.to_triplets();
// in Eigen?

是的，似乎没有这样的功能。

相反，

std::vector<Eigen::Triplet<double>> to_triplets(Eigen::SparseMatrix<double> & M){
    std::vector<Eigen::Triplet<double>> v;
    for(int i = 0; i < M.outerSize(); i++)
        for(typename Eigen::SparseMatrix<double>::InnerIterator it(M,i); it; ++it)
            v.emplace_back(it.row(),it.col(),it.value());
    return v;
}
auto t = to_triplets(SparseMat);

如果你想做得更快，可以在IDE中打开它，四处寻找指向数据数组的指针，并编写一个对运行时没有影响的复杂函数，因为矩阵是稀疏的，而复制在非零元素方面是线性的。

如教程所示：

#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 << "RowtColtVal" <<endl;
    for (int k=0; k < A.outerSize(); ++k)
    {
        for (SparseMatrix<int>::InnerIterator it(A,k); it; ++it)
        {
            cout << 1+it.row() << "t"; // row index
            cout << 1+it.col() << "t"; // col index (here it is equal to k)
            cout << it.value() << endl;
        }
    }
    return 0;
}
相关文章：