删除文件的第一行并将其插回输出中

时间:2010-12-30 18:55:23

标签: c++ iostream stringstream

我有一个csv文件,我按如下方式读取文件:

#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;

typedef vector <double> record_t;
typedef vector <record_t> data_t;
data_t data;

istream& operator >> ( istream& ins, record_t& record )
  {
  record.clear();

  string line;
  getline( ins, line );

  // Using a stringstream to separate the fields out of the line
  stringstream ss( line );
  string field;
  while (getline( ss, field, ',' ))
    {
    // for each field we wish to convert it to a double
    stringstream fs( field );
    double f = 0.0;  // (default value is 0.0)
    fs >> f;

    // add the newly-converted field to the end of the record    record.push_back( f );
    }
  return ins;
  }

//-----------------------------------------------------------------------------
istream& operator >> ( istream& ins, data_t& data )
  {
  data.clear();

  record_t record;
  while (ins >> record)
    {
    data.push_back( record );
    }
  return ins;
  }
//----------------------------------------------
int main() {
  ifstream infile( "2010.csv" );
  infile >> data;

  if (!infile.eof())
    {
    cout << "Error with the input file \n";
    return 1;
    }

  infile.close();

  //do something with "data"

  // write the data to the output.
}

现在我正在使用的文件就像

A,B,c,D,E,F
1,1,1,1,1,1,
2,2,2,2,2,2,
3,3,3,3,3,3,

因此,如果标题不存在,程序运行正常。如何删除标题并将其插回输出文件,如何保持相同的格式?

我从某个地方调整了这个代码,我不记得来源了。

2 个答案:

答案 0 :(得分:2)

只需在第一行和while循环中使用另一个字符串,将第一行视为特殊情况(跳过所有其他行的正常处理)。

答案 1 :(得分:2)

首先读取第一行,然后将流缓冲区放入此函数中? 看起来你不想改变这个功能。

ifstream infile( "2010.csv" );
string header;
std::getline(infile, header);
infile >> data;