用 c++ 解析 CSV

Question

我正在尝试创建一个代码，该代码将通过带有股票信息的 csv 数据库进行解析。目前，我生成了代码，以便使用关键字搜索并打印整行，但我正在尝试获取它，以便它以格式整齐的方式打印整行和标题行。

我正在尝试获取它，以便如果我搜索 Google，它会返回

符号 GOOG
名称 Google Inc
今日高点 $568.77

csv 的样子：

符号、名称、价格、今日最高价、今日最低价、52 周最低价
GOOG,Google Inc.,$568.77 ,$570.25 ,$560.35
AAPL，苹果公司，93.28 美元，63.89 美元，99.44 美元。

代码：

string NameSearch::getInput()
{
    cout << "Enter the name of the company you would like to search for: ";
    getline(cin, input);
    return input;

}
void NameSearch::NameAlgorithm()
{
    string line;
    ifstream fs("Stock Database.csv");

    while (!fs.eof())
    {
        getline(fs, line);
        string companyname = "";    
        string a;
        int column = 1;             
        int commacount = 0;         
        int ChrCount = 0;               

        while (line != "\0")        
        {
            a = line[ChrCount];       
            ChrCount++;

            if (a == ",")
            {
                commacount++; 
            }

            else if (commacount == column) 
            {
                companyname.append(a);
            }

            else if (commacount > column) 
            {
                break;
            }

            if (companyname == input)
            {
                cout << endl << line;
            }
        }
    }
}

Answer 1

首先应将逗号解析为空格。您可以通过更改流语言环境中的内部 std::ctype<charT> 方面来做到这一点：

struct csv_classification : std::ctype<char> {
    csv_classification() : ctype(make_table()) { }
private:
    static mask* make_table() {
        const mask* classic = classic_table();
        static std::vector<mask> v(classic, classic + table_size);
        v[','] |= space;
        v[' '] &= ~space;
        return &v[0];
    }
};

然后使用以下方法设置语言环境：

ifs.imbue(std::locale(ifs.getloc(), new csv_classification));

接下来制作一个操纵器来检查您是否在行尾。如果是，它会在流状态中设置std::ios_base::failbit 标志。还使用内部存储来判断记录是否属于映射中的键或值。向 Dietmar 借一点……

static int row_end = std::ios_base::xalloc();

std::istream& record(std::istream& is) {
    while (std::isspace(is.peek())) {
        int c(is.peek());
        is.ignore();

        if (c == '\n') {
            is.iword(row_end) = !is.iword(row_end);
            is.setstate(std::ios_base::failbit);
        }
    }
    return is;
}

那么你可以这样做：

std::vector<std::string> keys, values;

for (std::string item;;) {
    if (ifs >> record >> item)
        keys.push_back(item);
    else if (ifs.eof())
        break;
    else if (ifs.iword(row_end)) {
        ifs.clear();
        while (ifs >> record >> item)
            values.push_back(item);
    }
    else
        break;
}

现在我们需要应用键和值并将它们打印出来。我们可以为此创建一个新算法：

template<class Iter1, class Iter2, class Function>
void for_each_binary_range(Iter1 first1, Iter1 last1,
                           Iter2 first2, Iter2 last2, Function f)
{
    assert(std::distance(first1, last1) <= std::distance(first2, last2));

    while (first1 != last1) {
        f(*first1++, *first2++);
    }
}

我们终于做到了：

for_each_binary_range(std::begin(keys),   std::end(keys),
                      std::begin(values), std::end(values),
[&] (std::string const& key, std::string const& value)
{
    std::cout << key << ": " << value << std::endl;
}

Live Demo

Answer 2

这是您的解决方案（最接近您的要求）：

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

using namespace std;

typedef std::vector<std::string> record;

std::istream&
operator>>(
  std::istream& is,
  record&       r)
{
  r.clear();

  string line;
  getline(is, line);

  istringstream iss(line);
  string field;

  while (getline(iss, field, ',' ))
    r.push_back(field);

  return is;
}

int
main()
{

  ifstream file("Stock Database.csv");
  record headers, r;

  if (file.good())
    file >> headers;

  while (file >> r)
  {
    for (int i = 0; i < r.size(); i++)
      cout << headers[i] <<  ":\t" << r[i] << endl;

    cout << "------------------------------" << endl;
  }

  return 0;
}

// EOF

“Stock Database.csv”文件的内容是：

Symbol,Name,Price,High Today,Low Today,52 Week Low
GOOG,Google Inc.,$568.77 ,$570.25 ,$560.35
AAPL,Apple Inc.,$93.28 ,$63.89 ,$99.44

随心所欲地处理唱片。从文件中读取的第一次假设会为您带来标题。每次下一次读取都会用 csv 值填充每条记录。