C++讀取csv表格文件到vector

這個CSV文件假設知道每行有多少個數，也知道數據的格式，即可使用下面簡單的方法實現。

我們假設每行有4個數據，依次是int,int,float,float

基本思路是：把每行的數據定為一個類型，放在vector中，方便查找使用。使用FILE讀文件，fscanf快速把已知格式的數據讀進來。

 

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

using namespace std;

class csvdata{
public:
    int id;
    int level;
    float price;
    float cost;
};//行的類定義

int main()
{
    vector<csvdata> incsv;
    csvdata intp;
    FILE *fp;
    fp=fopen("D:\\111.csv","r");//你自己的文件路徑
    while(1){
        fscanf(fp,"%d,%d,%f,%f",&intp.id,&intp.level,&intp.price,&intp.cost);
        incsv.push_back(intp);
        if (feof(fp))break;
    }
    fclose(fp);
    for(int i=0;i<incsv.size();i++)
    {
        cout<<incsv[i].id<<" "<<incsv[i].level<<" "<<incsv[i].price<<" "<<incsv[i].cost<<endl;
        
    }//輸出顯示每行的數據


system("pause");
return 0;
}

 

如果不知道csv每行的個數和格式類型，也可以統一用string來讀取，只是比上面方法稍麻煩。

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

using namespace std;

string Trim(string& str)
{
 str.erase(0,str.find_first_not_of(" \t\r\n"));

 str.erase(str.find_last_not_of(" \t\r\n") + 1);

 return str;
}

int main()
{
 ifstream fin("test.csv");

 string line; 
 while (getline(fin, line)) {
  //cout << line << endl;

  istringstream sin(line); 
  vector<string> fields; 
  string field;
  while (getline(sin, field, ',')) {
   fields.push_back(field); 
  }

  string name = Trim(fields[0]); 
  string age = Trim(fields[1]); 
  string birthday = Trim(fields[2]); 
  cout << name << "\t" << age << "\t" << birthday << endl;
 }
    system("pause");
    return 0;
    
}