c++ - 将 vector<vector<double>> 转换为特征矩阵？

Question

假设我有一个 double vector 的 C++ vector ，vector<vector<double> > .此变量通常表示 CSV 文件的内容，其中文件中的每一行都是 double 的序列。值。每行都有相同的项目数。

转换这个 vector<vector<double> > 最优雅的方法是什么？变成二维特征矩阵Xd？

我希望像下面这样，我想从 double vector 的 vector 创建一个 3x5 矩阵。

#include "Eigen/Dense"
#include <vector>

using Eigen::MatrixXd;

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

    std::vector<double> row1 = {  1.0,  2.0,  3.0,  4.0,  5.0 };
    std::vector<double> row2 = { 11.0, 12.0, 13.0, 14.0, 15.0 };
    std::vector<double> row3 = { 21.0, 22.0, 23.0, 24.0, 25.0 };

    std::vector<std::vector<double> > vv;
    vv.push_back(row1);
    vv.push_back(row2);
    vv.push_back(row3);

    MatrixXd mymatrix(vv); // NOPE! Compiler error.

    return 0;
}

编译结果:

% g++ -std=c++11 test_eigen.cpp

test_eigen.cpp:85:14: error: no matching constructor for initialization of 'MatrixXd' (aka
      'Matrix<double, Dynamic, Dynamic>')
    MatrixXd matrix(vv); // NOPE!
             ^      ~~

请注意，在 Python 中，从列表列表转换为 Numpy 矩阵展示了我正在寻找的简单性:

import numpy as np

row1 = [  1.0,  2.0,  3.0,  4.0,  5.0 ]
row2 = [ 11.0, 12.0, 13.0, 14.0, 15.0 ]
row3 = [ 21.0, 22.0, 23.0, 24.0, 25.0 ]
list_of_list = [row1, row2, row3];

mymatrix = np.array(list_of_list)

相关问题，但使用一维 vector :

Initialise Eigen::vector with std::vector

Answer 1

感谢 user20650，我发现了一个重复的问题。我将其解决方案概括为可重用函数:

// Convert a 2-D vector<vector<double> > into an Eigen MatrixXd.
// Throws exception if rows do not have same length.
MatrixXd convert_vvd_to_matrix(vector<vector<double> > vvd) {

    std::size_t n_rows = vvd.size();
    std::size_t n_cols = vvd.at(0).size();

    MatrixXd result(n_rows, n_cols);
    result.row(0) = VectorXd::Map(&vvd[0][0], n_cols);

    // Add each vector row to the MatrixXd. 
    for (std::size_t i = 1; i < n_rows; i++) {

        // Make sure that every row of vvd has the same size.
        if (n_cols != vvd.at(i).size()) {
            char buffer[200];
            snprintf(buffer, 200, 
                        "vvd[%ld] size (%ld) does not match vvd[0] size (%ld)",
                        i, vvd.at(i).size(), n_cols);
            string err_mesg(buffer);
            throw std::invalid_argument(err_mesg);
        }

        result.row(i) = VectorXd::Map(&vvd[i][0], n_cols);
    }

    return result;
}

示例用法:

#include <cstdio>
#include <iostream>
#include "Eigen/Dense"
#include <vector>

using std::cout;
using std::endl;
using std::vector;
using Eigen::MatrixXd;

void test_vector_vector_double() {

    vector<double> row1 = {  1.0,  2.0,  3.0,  4.0,  5.0 };
    vector<double> row2 = { 11.0, 12.0, 13.0, 14.0, 15.0 };
    vector<double> row3 = { 21.0, 22.0, 23.0, 24.0, 25.0 };

    vector < vector<double> > vvd;
    vvd.push_back(row1);
    vvd.push_back(row2);
    vvd.push_back(row3);

    MatrixXd mat = convert_vvd_to_matrix(vvd);

    cout << mat << endl;
}

输出:

 1  2  3  4  5
11 12 13 14 15
21 22 23 24 25