博客

在做机器学习的时候，数据集经常以csv文件的格式保存，因此需要读取数据，以下是读取csv的代码以及一些关键函数的详解链接：

#include <iostream>
#include<string>
#include<vector>
#include<fstream>
#include<sstream>

using namespace std;

int main() {
  ifstream inFile("D:\\aa\\bb.csv");
  string lineStr;
  vector<vector<string>> strArray;
  while (getline(inFile, lineStr)) {
    cout << lineStr << endl;
    stringstream ss(lineStr);
    string str;
    vector<string> lineArray;
    while (getline(ss, str, ',')){
		lineArray.push_back(str);
	}
    strArray.push_back(lineArray);
  }
}

以下是一些关键函数的详解：

ofstream和ifstream详细用法

getline()用法详解

stringstream用法详解以及和getline配合分割字符串

如果想把string转化成float输出，可用下方代码：

#include <iostream>
#include<string>
#include<vector>
#include<fstream>
#include<sstream>

using namespace std;

int main() {
  ifstream inFile("D:\\LaneChangeDecision\\lc_data\\train_data_v0719.csv");
  string lineStr;
  vector<vector<float>> strArray;
  while (getline(inFile, lineStr)) {
    //cout << lineStr << endl;
    stringstream ss(lineStr);
    string str;
    vector<float> lineArray;
    while (getline(ss, str, ',')){
      \\string转float
      float str_float;
      istringstream istr(str);
      istr >> str_float;
	  lineArray.push_back(str_float);
	}
    strArray.push_back(lineArray);
  }
}