当前位置:  开发笔记 > 编程语言 > 正文

如何读取CSV文件并分配给Eigen Matrix?

如何解决《如何读取CSV文件并分配给EigenMatrix?》经验,为你挑选了1个好方法。

我尝试将一个大的cvs文件读入Eigen Matrix,在发现有问题的代码下面,它无法检测cvs文件中的每行\n在矩阵中创建多行.(它用单行读取整个文件).不确定代码有什么问题.谁能在这里建议?我也在寻找一种有效的方法来读取10k行和1k cols的csv文件.不太确定下面的代码是最有效的方法吗?非常感谢你的评论.

#include 
#include 
#include 
#include 
#include  //DataFile.fail()  function
#include 
#include 
#include 
using namespace std;


#include 
#include 
using namespace Eigen;

 void readCSV(istream &input, vector< vector > &output)
{
    int a = 0;
    int b = 0;

    string csvLine;
    // read every line from the stream
    while( std::getline(input, csvLine) )
    {

        istringstream csvStream(csvLine);
        vector csvColumn;
        MatrixXd mv;
        string csvElement;
        // read every element from the line that is seperated by commas
        // and put it into the vector or strings
        while( getline(csvStream, csvElement, ' ') )
        {
            csvColumn.push_back(csvElement);
            //mv.push_back(csvElement);
            b++;
        }       
        output.push_back(csvColumn);
        a++;
    }
    cout << "a : " << a << " b : " << b << endl;   //a doen't detect '\n'
}

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

    cout<< "ELM" << endl;
    //Testing to load dataset from file.
    fstream file("Sample3.csv", ios::in);
    if(!file.is_open())
    {
        cout << "File not found!\n";
        return 1;
    }
    MatrixXd m(3,1000);
    // typedef to save typing for the following object
    typedef vector< vector > csvVector;
    csvVector csvData;

    readCSV(file, csvData);
    // print out read data to prove reading worked
    for(csvVector::iterator i = csvData.begin(); i != csvData.end(); ++i)
    {
        for(vector::iterator j = i->begin(); j != i->end(); ++j)
        {
           m(i,j) = *j; 
           cout << *j << ", ";
        }
        cout << "\n";
    }
}

我还将附上一个示例cvs文件.https://onedrive.live.com/redir?resid=F1507EBE7BF1C5B!117&authkey=!AMzCnpBqxUyF1BA&ithint=file%2ccsv



1> user357269..:

这是你可以实际复制粘贴的东西

编写自己的"解析器"

优点:轻巧且可定制

缺点:可定制

#include 
#include 
#include 

using namespace Eigen;

template
M load_csv (const std::string & path) {
    std::ifstream indata;
    indata.open(path);
    std::string line;
    std::vector values;
    uint rows = 0;
    while (std::getline(indata, line)) {
        std::stringstream lineStream(line);
        std::string cell;
        while (std::getline(lineStream, cell, ',')) {
            values.push_back(std::stod(cell));
        }
        ++rows;
    }
    return Map>(values.data(), rows, values.size()/rows);
}

用法:

MatrixXd A = load_csv("C:/Users/.../A.csv");
Matrix3d B = load_csv("C:/Users/.../B.csv");
VectorXd v = load_csv("C:/Users/.../v.csv");

使用armadillo库的解析器

优点:支持其他格式,而不仅仅是csv

缺点:额外的依赖

#include 

template 
M load_csv_arma (const std::string & path) {
    arma::mat X;
    X.load(path, arma::csv_ascii);
    return Eigen::Map(X.memptr(), X.n_rows, X.n_cols);
}

推荐阅读
ar_wen2402851455
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有