python读写csv文件方法详细总结
python提供了大量的库,可以非常方便的进行各种操作,现在把python中实现读写csv文件的方法使用程序的方式呈现出来。
在编写python程序的时候需要csv模块或者pandas模块,其中csv模块使不需要重新下载安装的,pandas模块需要按照对应的
python版本安装。
在python2环境下安装pandas的方式是:
sudopipinstallpandas
在python3环境下安装pandas的方式是:
sudopip3installpandas
1、使用csv读写csv文件方法总结
读文件的时候,打开文件,调用csv.reader()读取文件;对于读取之后的文件的内容,要把这些内容输入到另一个文件中保存,可以通过遍历读取的文件的每一行,然后使用csv_write.writerow()的方式写入到指定的文件。
2、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv #读取csv文件 csv_file=csv.reader(open('../../data/capital/2010-Q4-cabi-trip-history-data.csv')) print(csv_file) #添加newline可以避免一行之后的空格,这样需要在python3环境下运行 out=open('../../data/capital/2010-Q4','a',newline='') csv_write=csv.writer(out,dialect='excel') foritemincsv_file: #printitem csv_write.writerow(item) print("writeover")
上面的代码是传统的打开文件---读取文件---打开文件---关闭文件的方式进行文件读取和写入。在读取和写入的过程中分别调用了csv库的reader()和writer()
3、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv #下面这种加了encoding的写法需要在python3环境下运行 csv_reader=csv.reader(open('../../data/capital/2011-Q1-cabi-trip-history-data.csv',encoding='utf-8')) out=open('../../data/capital/a0','w',newline='') csv_writer=csv.writer(out,dialect='excel') forrowincsv_reader: #输出的每一行是一个list,list中的每一个元素转换成了string类型 print(row) csv_writer.writerow(row)
示例代码2的读写方式与示例代码1的方式基本相同,不同的是在示例代码2中打开文件的时候指定了打开时的编码方式,对于输出文件的打开方式也与示例代码1的方式有少量的差别,把写入方式右'a'改成了'w
4、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv #读取csv文件方式1 csvFile=open('../../data/capital/2011-Q3-cabi-trip-history-data.csv','r') reader=csv.reader(csvFile) data=[] foriteminreader: # print(item) data.append(item) #print(data) csvFile.close() #读取csv文件方式2 withopen("../../data/capital/2011-Q3-cabi-trip-history-data.csv",'r')ascsvFile: #读取csv文件,返回的是迭代类型 reader2=csv.reader(csvFile) foritem2inreader2: print(item2) csvFile.close() #从列表写入csv文件 #设置newline,否则两行之间会空一行 csvFile2=open('../../data/capital/0001.csv','w',newline='') writer=csv.writer(csvFile2) m=len(data) foriinrange(m): writer.writerow(data[i]) csvFile2.close() #从字典写入csv文件 dic={'张三':123,'李四':456,'王二娃':789} csvFile3=open('../../data/capital/0001.csv','w',newline='') writer2=csv.writer(csvFile3) forkeyindic: print(key) # writer2.writerow([key,dic[key]]) csvFile3.close()
在上面的代码中提出了两种不同的打开csv文件和写入csv文件的方式。其中第一种打开方式与前面两种非常相似,不同之处是在这种方式中,打开文件的时候加入了读取方式,而前面两个示例代码没有加。另一种读取方式是采用python中独有的命名方式对打开文件命名,其他部分相似。在这个示例中,对列表和字典写入csv文件做了区分,分别提供了列表和字典写入csv文件的示例代码。
5、使用csv读写csv文件示例代码
#encoding:utf-8 ''' importpandasaspd #任意的多组列表 a=[1,2,3] b=[4,5,6] #字典中的key值即为csv中的列名 dataFrame=pd.DataFrame({'a_name':a,'b_name':b}) #将DataFrame存储为csv,index表示是否显示行名,default=True dataFrame.to_csv('0001.csv',index=False,sep='') #pandas提供的读取csv的方法 data=pd.read_csv('test.csv') ''' #=================================== #另一种方法:用csv包,一行一行写入 importcsv #python2可以用file替代open withopen('test.csv','w')ascsvFile: writer=csv.writer(csvFile) #先写columns_name writer.writerow(["index","a_name","b_name"]) #写入多行用writerows writer.writerows([[1,2,3],[0,1,2],[4,5,6]]) #用reder读取csv文件 withopen('test.csv','r')ascsvFile: reader=csv.reader(csvFile) forlineinreader: printline
这个示例代码提出了写入一行和写入多行的实现方式。
6、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv withopen('test.csv')ascsvFile: #readcsv=csv.reader(csvFile,delimiter='') readcsv=csv.reader(csvFile) # rows=[rowforrowinreadcsv] forrowinreadcsv: rows=[rowforrowinreader] #print(row)#打印一行 print(row[0])#打印一行中的某个cell print(row[0],row[1])#打印一行中的某个cell print('=================') withopen('test.csv')ascsvFile: readCSV=csv.reader(csvFile) cols1=[] cols2=[] cols3=[] forrowinreadCSV: col1=row[0] col2=row[1] col3=row[2] cols1.append(col1) cols2.append(col2) cols3.append(col3) printcols1 printcols2 printcols3
上面的代码示例提出了读取所有行的新写法,提出了读取某个cell和某一列的方式。
7、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv #第一种方法:使用reader函数,接收一个可迭代的对象(比如csv文件),能返回一个生成器,就可以从其中解析出csv的内容.eg:下面的代码可以读取csv的全部内容,以行为单位 withopen('test.csv','rb')ascsvFile: reader=csv.reader(csvFile) rows=[rowforrowinreader] printrows print('===========') #提取其中某一列可以使用下面的方法 withopen('test.csv','rb')ascsvFile: reader=csv.reader(csvFile) column=[row[1]forrowinreader] printcolumn print('======第二种方法======') #第二种方法:使用DictReader,和reader函数类似,接收可迭代的对象,能返回一个可迭代的对象,能返回一个生成器,但是返回的每一个单元格都放在一个字典的值内,而这个字典的键则是这个单元格的标题.用下面的代码可以看到DictReader的结构: withopen('test.csv','rb')ascsvFile: reader=csv.DictReader(csvFile) column=[rowforrowinreader] print(column) #使用CictReader读取csv的某一列,这样可以按照列的标题查询 withopen('test.csv','rb')ascsvFile: reader=csv.DictReader(csvFile) column=[row['a_name']forrowinreader] printcolumn
上面的代码提出了读取整行和整列的新方式。另外提出了一种使用DictReader的方式按照列名读取的方式。
8、使用csv读写csv文件示例代码
#encoding:utf-8 importcsv withopen('test.csv','rb')ascsvFile: reader=csv.reader(csvFile) forrowinreader: printrow #向csv文件中写内容,这样写会把内容全部写到一个cell中 withopen('test1.csv','wb')ascsvFile: writer=csv.writer(csvFile,delimiter='',quotechar='|',quoting=csv.QUOTE_MINIMAL) writer.writerow(['a','4','5','6']) writer.writerow(['b','1','2','3']) writer.writerow(['c','9','8','7']) writer.writerow(['d','8','3','1']) #向csv文件中写内容,下面的写法最终可以把每一列的内容放在一个cell中 withopen('test2.csv','wb')ascsvFile: writer=csv.writer(csvFile,dialect='excel') writer.writerow(['a','4','5','6']) writer.writerow(['b','1','2','3']) writer.writerow(['c','9','8','7']) writer.writerow(['d','8','3','1'])
上面的代码提出了可以把列表中的数据放入到不同cell的实现方式。