自定函數實現C++讀取CSV文件

C++是一個應用方式非常靈活的編程語言。我們可以通過自己的需求，自己編寫一個適合自己開發的函數來實現一些特定的功能。比如今天為大家介紹的就是自定義函數來實現C++讀取CSV文件這一功能。

• C++靜態構造函數相關概念解析
• C++ Traits應用技巧分享
• C++虛函數示例解讀
• C++函數指針與C#委托之間有何聯系
• C++ explicit關鍵字應用於構造函數

今天恰好做一個東西需要一個C++讀取CSV文件的函數。

沒有找到現成的函數，所以就自己寫了一個。

目的就是實現C++讀取CSV文件，每一行讀到一個數組中。

有點生疏了，字符串處理函數都不知道有什麼現成的，所以也寫了兩個輔助的字符串處理函數。

第一個函數：計算一個字符串中的某個子串的個數

int strstr_cnt(const char *string, const char 
*substring) {  
int i,j,k,count = 0;  
for (i = 0; string[i]; i++){  
for (j = i, k = 0; (string[j] == 
substring[k] && (j < strlen(string))); j++,k++) {  
if (! substring[k + 1]) {  
count++;  
}  
}  
}  
return count;  
} 

第二個函數：計算一個子串在字符串中的位置

int substring_index(const char *s1,const char *s2, int pos){  
int i,j,k;  
for( i = pos ; s1[i] ; i++ ) {  
for( j = i, k = 0 ; s1[j] == s2[k]; j++,k++ ){  
if (! s2[k + 1]) {  
return i;  
}  
}  
}  
return -1;  
} 

第三個函數：讀取已經打開的一個CSV文件的一行，將這一行處理到一個數組中。

char *fgetcsvline(vector<string> &csv_databuf, FILE *fhead) {  
char *ret_stat;  
char data_buf[1024];  
string stringbuf;  
ret_stat = fgets( data_buf, 1024, fhead );  
if (ret_stat != NULL) {  
int len = strstr_cnt(data_buf,"\",\"");  
if (len > 0){  
int pos = substring_index(data_buf,"\",\"",0);  
int startpos = 1;  
string csv_buf;  
while (pos > 0) {  
stringbuf = (string)data_buf;  
csv_buf = stringbuf.substr(startpos,pos - startpos);  
csv_databuf.push_back(csv_buf);  
startpos = pos + 3;  
pos = substring_index(data_buf,"\",\"",pos + 2);  
}  
if ((substring_index(data_buf,"\n",0)) > 0){  
csv_buf = stringbuf.substr(startpos, stringbuf.length() 
- startpos - 2);  
} else {  
csv_buf = stringbuf.substr(startpos, stringbuf.length() 
- startpos - 1);  
}  
csv_databuf.push_back(csv_buf);  
}  
}  
return ret_stat;   
} 

這個函數使用上面的兩個函數來處理字符串。

另外這個函數用來處理的CSV文件是帶雙引號格式的：

"sss","ddd","444"  
"ttt","www","ooo"  
"sss","qqq","000" 

使用方法如下：

int main(int argc, char* argv[]) {  
FILE *fp_head;  
string csvFileName = "test.csv";  
char *ret_stat;  
vector<string> csv_data;  
fp_head = fopen( csvFileName, "rt" );  
ret_stat = fgetcsvline(csv_data, fp_head);  
while (ret_stat != NULL) {  
//get csv data use csv_data[n]  
ret_stat = fgetcsvline(csv_data, fp_head);  
}  
return 0;  
} 

以上代碼可能需要稍加調試。也可稍加改動用來讀取其它格式的csv文件。

以上就是C++讀取CSV文件的相關實現方法。