快捷導(dǎo)航

C++中實(shí)現(xiàn)保存數(shù)據(jù)到CSV文件

更新時(shí)間：2023年08月04日 10:21:35 作者：熊叫大雄

這篇文章主要介紹了C++中實(shí)現(xiàn)保存數(shù)據(jù)到CSV文件方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

C++保存數(shù)據(jù)到CSV文件

主要是今天工作的時(shí)候需要把一些數(shù)據(jù)保存到本地，因?yàn)槭且恍╊A(yù)測(cè)值和標(biāo)簽的對(duì)比，還有預(yù)測(cè)值的概率，所以想到用CSV文件來(lái)保存，大概查了一下，還是比較簡(jiǎn)單的，所以記錄一下。

首先要說(shuō)明的是CSV文件有點(diǎn)類(lèi)似excel文件，也可以用excel文件直接打開(kāi)的。在寫(xiě)數(shù)據(jù)到CSV文件的時(shí)候要注意，字符串逗號(hào)表示的是換列，換行符號(hào)就是換行，所以在寫(xiě)數(shù)據(jù)的時(shí)候要注意好這兩點(diǎn)就可以了。

然后寫(xiě)的時(shí)候也很簡(jiǎn)單，直接用C++的ofstream就可以實(shí)現(xiàn)了，ofstream有一個(gè)操作是"<<"，這個(gè)也很好用，就類(lèi)似std::cout的操作一樣即可。

比如我現(xiàn)在的數(shù)據(jù)是比較統(tǒng)一的，每一個(gè)樣本是一行，一行數(shù)據(jù)要分成四列，第一列是樣本的圖像地址，第二列是標(biāo)簽，第三列是最終預(yù)測(cè)值，第四列是概率，每一列的格式是一樣的，那么我的代碼就是這樣：

ofstream file(CSV_PATH);
if (file)
{
? ? file << image_path << "," << label << "," << prediction << "," << probability << "\n";
}
file.close();

這樣就可以實(shí)現(xiàn)了，還是比較容易的。

C++對(duì)csv文件操作（讀、寫(xiě)、追加）

使用說(shuō)明：csv文件按照","進(jìn)行分隔。因此每個(gè)內(nèi)容中需避免出現(xiàn)","

1.讀csv文件

c++通過(guò)文件讀入方式打開(kāi)文件。即通過(guò)ifstream類(lèi)進(jìn)行打開(kāi)文件。

 
string fname = "test.csv";
//以讀入方式打開(kāi)文件
ifstream csv_data(fname, ios::in);
if (!csv_data.is_open())
{
	cout << "Error: opening file fail" << endl;
	exit(1);
}
else {
	string line;
	vector<string> words; //聲明一個(gè)字符串向量
	string word;
	// ------------讀取數(shù)據(jù)-----------------
	// 讀取標(biāo)題行
	getline(csv_data, line);
	istringstream sin;
	// 按行讀取數(shù)據(jù)
	while (getline(csv_data, line))
	{
		// 清空vector及字符串流,只存當(dāng)前行的數(shù)據(jù)
		words.clear();
		sin.clear();
		sin.str(line);
		//將字符串流sin中的字符讀到字符串?dāng)?shù)組words中，以逗號(hào)為分隔符
		while (getline(sin, word, ','))
		{
			cout << word << endl;
			words.push_back(word); //將每一格中的數(shù)據(jù)逐個(gè)push
		}
	}
	csv_data.close();
}

2.寫(xiě)入csv文件

c++通過(guò)文件寫(xiě)入方式打開(kāi)文件進(jìn)行寫(xiě)入。即通過(guò)ofstream類(lèi)進(jìn)行寫(xiě)入，并在打開(kāi)文件中指明ios::out。

說(shuō)明：默認(rèn)通過(guò)iso::out方式進(jìn)行寫(xiě)入，當(dāng)文件不存在時(shí)會(huì)進(jìn)行創(chuàng)建

string fname = "test.csv";
ofstream outFile(fname, ios::out);
// 寫(xiě)入標(biāo)題行
outFile << "name" << ','
<< "income" << ','
<< "expenditure" << ','
<< "addr" << endl;
// ********寫(xiě)入兩行數(shù)據(jù)*********
outFile << "zhangsan" << ','
<< "3000" << ','
<< "1200" << ','
<< "陜西省" << endl;
outFile << "lisi" << ','
<< to_string(2032.1) << ','
<< to_string(789.2) << ','
<< "北京市" << endl;
//數(shù)字需轉(zhuǎn)為字符串進(jìn)行寫(xiě)入,csv文件結(jié)束一行寫(xiě)入需要"\n"或者endl進(jìn)行換行
outFile.close();

3.向csv文件中追加內(nèi)容

與第2部分幾乎相同，只不過(guò)是打開(kāi)文件時(shí)選擇ios::app方式進(jìn)行。當(dāng)文件不存在時(shí)會(huì)進(jìn)行創(chuàng)建

ofstream outFile(fname, ios::app);
// ********寫(xiě)入兩行數(shù)據(jù)*********
outFile << "wangwu" << ','
	<< "1234" << ','
	<< to_string(12.32) << ','
	<< "河南省" << endl;
outFile << "lisi" << ','
	<< to_string(2032.1) << ','
	<< to_string(789.2) << ','
	<< "北京市" << endl;
//數(shù)字需轉(zhuǎn)為字符串進(jìn)行寫(xiě)入,csv文件結(jié)束一行寫(xiě)入需要"\n"或者endl進(jìn)行換行
outFile.close();

4.具體使用

4.1讀入csv文件

#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <string>
using namespace std;
void PrintCSVLine(vector<string> line_data)
{
	//此語(yǔ)法僅在C++11中適用
	for (string str : line_data)
	{
		cout << str << " ";
	}
	cout << endl;
}
//讀入csv文件
int main() {
	string fname = "test.csv";
	//以讀入方式打開(kāi)文件
	ifstream csv_data(fname, ios::in);
	if (!csv_data.is_open())
	{
		cout << "Error: opening file fail" << endl;
		exit(1);
	}
	else {
		string line;
		vector<string> words; //聲明一個(gè)字符串向量
		string word;
		// ------------讀取數(shù)據(jù)-----------------
		// 讀取標(biāo)題行
		getline(csv_data, line);
		istringstream sin;
		// 按行讀取數(shù)據(jù)
		while (getline(csv_data, line))
		{
			// 清空vector及字符串流,只存當(dāng)前行的數(shù)據(jù)
			words.clear();
			sin.clear();
			sin.str(line);
			//將字符串流sin中的字符讀到字符串?dāng)?shù)組words中，以逗號(hào)為分隔符
			while (getline(sin, word, ','))
			{
				//cout << word << endl;
				words.push_back(word); //將每一格中的數(shù)據(jù)逐個(gè)push
			}
			//輸出此行中的內(nèi)容
			PrintCSVLine(words);
		}
		csv_data.close();
	}
}

4.2寫(xiě)入csv文件

#include <iostream>
#include <fstream>
#include <string>
using namespace std;
int main()
{
	string fname = "test.csv";
	ofstream outFile(fname, ios::out);
	if (outFile.is_open())  // 檢查文件是否打開(kāi)成功
	{
		// 寫(xiě)入標(biāo)題行
		outFile << "name" << ','
			<< "income" << ','
			<< "expenditure" << ','
			<< "addr" << endl;
		// ********寫(xiě)入兩行數(shù)據(jù)*********
		outFile << "zhangsan" << ','
			<< "3000" << ','
			<< "1200" << ','
			<< "陜西省" << endl;
		outFile << "lisi" << ','
			<< to_string(2032.1) << ','
			<< to_string(789.2) << ','
			<< "北京市" << endl;
		//數(shù)字需轉(zhuǎn)為字符串進(jìn)行寫(xiě)入,csv文件結(jié)束一行寫(xiě)入需要"\n"或者endl進(jìn)行換行
		outFile.close();
	}
	else
	{
		cout << "文件無(wú)法打開(kāi)！" << endl;
	}
}

4.3向csv文件追加

#include <iostream>
#include <fstream>
#include <string>
using namespace std;
int main()
{
	string fname = "test.csv";
	//先判斷文件是否存在
	ifstream file(fname);
	if (!file.is_open()) {
		cout << "File does not exist!" << endl;
		return 1;
	}
	else {
		cout << "File exists!" << endl;
		file.close();//必須先關(guān)閉文件后才可寫(xiě)入
		ofstream outFile(fname, ios::app);
		// ********寫(xiě)入兩行數(shù)據(jù)*********
		outFile << "wangwu" << ','
			<< "1234" << ','
			<< to_string(12.32) << ','
			<< "河南省" << endl;
		outFile << "lisi" << ','
			<< to_string(2032.1) << ','
			<< to_string(789.2) << ','
			<< "北京市" << endl;
		//數(shù)字需轉(zhuǎn)為字符串進(jìn)行寫(xiě)入,csv文件結(jié)束一行寫(xiě)入需要"\n"或者endl進(jìn)行換行
		outFile.close();
	}
	return 0;
}