python123平台作业答案第十周csv文件编程_CSV文件的使用－写入／写出(C++/Python)

参考：

#########################################################################

CSV-逗号分隔符(Comma-Separatored Values，有时也可用字符分隔符，分隔字符不一定要使用逗号)

典型分隔符是逗号，分号或制表符

excel表格可以打开csv文件，如果分隔符使用的是逗号的话，表格可以按逗号分隔，但如果不是逗号的话，并不能分隔

新建一个文件test.csv(利用逗号分隔):

hello,world

hai,hi

利用excel表格打开

新建另一个文件test2.csv(利用分号分隔):

hello;world

hai;hi

利用excel表格打开：

并没有按分号分隔

所以如果我们想要读取表格数据时，可以先将表格保存为csv文件，然后利用分隔符','读取表格数据

有时我们想要在数据中使用分隔符这时可以用双引号：

test3.csv:

"adsf,asdfa",323

gadfasf,23

test4.csv:

"asdf'asdf',adf",32

"ads'asd',af",45

#########################################################33

C++实现

写入CSV文件：

#include

#include

#include

#include

#include

#include

#include

#include

#include

using namespace std;

//图片地址

const char *address="D:\\VSProject\\CSVTest\\CSVTest\\images\\";

//csv存储地址

const char *csv_addr="D:\\VSProject\\CSVTest\\CSVTest\\images.csv";

/**

* @function: 获取cate_dir目录下的所有文件名

* @param: cate_dir - const char指针类型

* @result：vector类型

*/

vector getFiles(const char *cate_dir)

{

vector files;//存放文件名

#ifdef WIN32

_finddata_t file;

long lf;

//输入文件夹路径

if ((lf=_findfirst(cate_dir, &file)) == -1) {

cout<

} else {

while(_findnext(lf, &file) == 0) {

//输出文件名

//cout<

if (strcmp(file.name, ".") == 0 || strcmp(file.name, "..") == 0)

continue;

files.push_back(file.name);

}

}

_findclose(lf);

#endif

#ifdef linux

DIR *dir;

struct dirent *ptr;

char base[1000];

if ((dir=opendir(cate_dir)) == NULL)

{

perror("Open dir error...");

exit(1);

}

while ((ptr=readdir(dir)) != NULL)

{

if(strcmp(ptr->d_name,".")==0 || strcmp(ptr->d_name,"..")==0) ///current dir OR parrent dir

continue;

else if(ptr->d_type == 8) ///file

//printf("d_name:%s/%s\n",basePath,ptr->d_name);

files.push_back(ptr->d_name);

else if(ptr->d_type == 10) ///link file

//printf("d_name:%s/%s\n",basePath,ptr->d_name);

continue;

else if(ptr->d_type == 4) ///dir

{

files.push_back(ptr->d_name);

/*

memset(base,'\0',sizeof(base));

strcpy(base,basePath);

strcat(base,"/");

strcat(base,ptr->d_nSame);

readFileList(base);

*/

}

}

closedir(dir);

#endif

//排序，按从小到大排序

sort(files.begin(), files.end());

return files;

}

/**

* @function: 将图片信息以及对应ID写入文件，保存为csv格式

* @param: images - const vector类型，图片地址

* @param: labels - const vector类型，图片ID

* @param: addr - const char指针类型，存储地址

* @param: separator - char类型，分隔符，默认值为';'

*/

bool write_csv(const vector images, const vector labels, const char *addr, char separator=';')

{

ofstream fout(addr);

if (!fout.is_open())

{

cout<

return false;

}

for (int i=0; i

{

fout<

fout<

fout<

}

fout.close();

return true;

}

int main()

{

//获取所有类别地址

string image_address=(string)address+"*";

vector cates=getFiles(image_address.c_str());

//图片地址

vector images;

//图片ID

vector labels;

for (int i=0; i

{

//获取类别绝对地址

string cate_dir=(string)address+cates[i];

//获取图片

string cate_addr=cate_dir+"\\*";

vector imgs=getFiles(cate_addr.c_str());

//判断图片是否存在

if ((int)imgs.size() == 0)

{

cout<

exit(1);

}

//设置绝对地址

for (int j=0; j

{

string img_dir=cate_dir+"\\"+imgs[j];

images.push_back(img_dir);

labels.push_back(i+1);

}

}

time_t start=clock();

//制作csv文件

write_csv(images, labels, csv_addr);

time_t end=clock();

double time=(double)(end-start)/CLOCKS_PER_SEC; //计算函数使用时间，以秒计

cout<

cout<

cin.get();

return 0;

}

读出CSV文件:

static void read_csv(const string& filename, vector& images, vector& labels, char separator = ';') {

std::ifstream file(filename.c_str(), ifstream::in);

if (!file) {

string error_message = "No valid input file was given, please check the given filename.";

CV_Error(CV_StsBadArg, error_message);

}

string line, path, classlabel;

while (getline(file, line)) {

stringstream liness(line);

getline(liness, path, separator);

getline(liness, classlabel);

if(!path.empty() && !classlabel.empty()) {

images.push_back(imread(path, 0));

labels.push_back(atoi(classlabel.c_str()));

}

}

}

#########################################################

python实现

#!/usr/bin/env python

#-*- coding: utf-8 -*-

'''

实现csv文件的读出，写入功能

'''

__author__ = 'zj'

import time

import os

#csv文件写入，默认分隔符是';'

#参数csv_name表示存储地址

#参数rows_list存储写入内容，是列表类型，每个元素代表一行输入内容，同样是列表类型

#例：rows_list=[[23,53,'123'], [434, 3, 'dfgv']]

def write_csv(csv_name, rows_list, separator=';'):

start=time.clock()

#打开文件

fi=open(csv_name, 'w')

#遍历每一行

for row in rows_list:

#遍历每一列

for col in row:

fi.write(str(col))

fi.write(separator)

#endfor

fi.write("\n")

#endfor

fi.close()

end=time.clock()

print 'write_csv function need time :%f s'%(end-start)

#end of write_csv

#csv文件读出，默认分隔符是';'

#参数csv_name表示存储地址

#参数rows_list存储写出内容，是列表类型，每个元素代表一行输入内容，同样是列表类型

#例：rows_list=[[23,53,'123'], [434, 3, 'dfgv']]

def read_csv(csv_name, rows_list, separator=';'):

start=time.clock()

#打开文件

fi=open(csv_name, 'r')

#读取每一行

for line in fi.readlines():

#分隔每一列

cols=line.split(separator)

rows_list.append(cols)

#endfor

fi.close()

end=time.clock()

print 'read_csv function need time :%f s'%(end-start)

#end of read_csv

if __name__ == '__main__':

#设置存储地址

address=os.path.join(os.getcwd(), "test.csv")

#设置存储内容

rows_list=[[23,53,'123'], [434, 3, 'dfgv']]

#写入csv文件

write_csv(address, rows_list, ';')

lists=[]

#读出csv文件

read_csv(address, lists, ';')

for line in lists:

for param in line:

print param+" ",

#endfor

print

#endfor

#endif