我有一个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,
因此,如果标题不存在,程序运行正常。如何删除标题并将其插回输出文件,如何保持相同的格式?
我从某个地方调整了这个代码,我不记得来源了。
答案 0 :(得分:2)
只需在第一行和while循环中使用另一个字符串,将第一行视为特殊情况(跳过所有其他行的正常处理)。
答案 1 :(得分:2)
首先读取第一行,然后将流缓冲区放入此函数中? 看起来你不想改变这个功能。
ifstream infile( "2010.csv" );
string header;
std::getline(infile, header);
infile >> data;