python讀寫csv檔案方法總結

轉自：https://blog.csdn.net/jp_666/article/details/78703068

python提供了大量的庫，可以非常友善的進行各種操作，現在把python中實作讀寫csv檔案的方法使用程式的方式呈現出來。

在編寫python程式的時候需要csv子產品或者pandas子產品，其中csv子產品使不需要重新下載下傳安裝的，pandas子產品需要按照對應的python版本安裝。

在python2環境下安裝pandas的方式是：

sudo pip install pandas
           

在python3環境下安裝pandas的方式是：

sudo pip3 install pandas
           

1、使用csv讀寫csv檔案方法總結

讀檔案的時候，打開檔案，調用csv.reader()讀取檔案；對于讀取之後的檔案的内容，要把這些内容輸入到另一個檔案中儲存，可以通過周遊讀取的檔案的每一行，然後使用csv_write.writerow()的方式寫入到指定的檔案。

2、使用csv讀寫csv檔案示例代碼1

#encoding:utf-8

import csv

#讀取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')

for item in csv_file:
	#print item
	csv_write.writerow(item)

print("write over")
           

上面的代碼是傳統的打開檔案---讀取檔案---打開檔案---關閉檔案的方式進行檔案讀取和寫入。在讀取和寫入的過程中分别調用了csv庫的reader()和writer()

3、使用csv讀寫csv檔案示例代碼2

#encoding:utf-8

import csv

#下面這種加了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')

for row in csv_reader:
	#輸出的每一行是一個list,list中的每一個元素轉換成了string類型
	print(row)
		csv_writer.writerow(row)
           

示例代碼2的讀寫方式與示例代碼1的方式基本相同，不同的是在示例代碼2中打開檔案的時候指定了打開時的編碼方式，對于輸出檔案的打開方式也與示例代碼1的方式有少量的差别，把寫入方式右’a’改成了’w’

4、使用csv讀寫csv檔案示例代碼3

#encoding:utf-8

import csv

#讀取csv檔案方式1

csvFile = open('../../data/capital/2011-Q3-cabi-trip-history-data.csv','r')

reader = csv.reader(csvFile)

data = []

for item in reader:
#	print(item)
	data.append(item) 

#print(data)

csvFile.close()

#讀取csv檔案方式2

with open("../../data/capital/2011-Q3-cabi-trip-history-data.csv",'r') as csvFile:
	#讀取csv檔案,傳回的是疊代類型
	reader2 = csv.reader(csvFile)
	for item2 in reader2:
		print(item2)

csvFile.close()

#從清單寫入csv檔案

#設定newline,否則兩行之間會空一行

csvFile2 = open('../../data/capital/0001.csv','w',newline='')

writer = csv.writer(csvFile2)

m = len(data)

for i in range(m):
	writer.writerow(data[i])

csvFile2.close()

#從字典寫入csv檔案

dic = {'張三':123,'李四':456,'王二娃':789}

csvFile3 = open('../../data/capital/0001.csv','w',newline='')

writer2 = csv.writer(csvFile3)

for key in dic:
	print(key)
#	writer2.writerow([key,dic[key]]) 

csvFile3.close()
           

在上面的代碼中提出了兩種不同的打開csv檔案和寫入csv檔案的方式。其中第一種打開方式與前面兩種非常相似，不同之處是在這種方式中，打開檔案的時候加入了讀取方式，而前面兩個示例代碼沒有加。另一種讀取方式是采用python中獨有的命名方式對打開檔案命名，其他部分相似。在這個示例中，對清單和字典寫入csv檔案做了區分，分别提供了清單和字典寫入csv檔案的示例代碼。

5、使用csv讀寫csv檔案示例代碼4

#encoding:utf-8

import pandas as pd

#任意的多組清單

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包,一行一行寫入

import csv 

#python2可以用file替代open
with  open('test.csv','w') as csvFile:
	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檔案
with open('test.csv','r') as csvFile:
	reader = csv.reader(csvFile)
	for line in reader:
		print line
           

這個示例代碼提出了寫入一行和寫入多行的實作方式。

6、使用csv讀寫csv檔案示例代碼5

#encoding:utf-8

import csv

with open('test.csv') as csvFile:
	#readcsv = csv.reader(csvFile,delimiter='')
	readcsv = csv.reader(csvFile)
	#
	rows= [row for row in readcsv]
	for row in readcsv:
		rows= [row for row in reader]
		#print(row)#列印一行
		print(row[0])#列印一行中的某個cell
		print(row[0],row[1])#列印一行中的某個cell 

print('=================')

with open('test.csv') as csvFile:
	readCSV = csv.reader(csvFile)
	cols1 = []
	cols2 = [] 
	cols3 = []
	for row in readCSV:
		col1 = row[0]	
		col2 = row[1]
		col3 = row[2]
		cols1.append(col1)
		cols2.append(col2)
		cols3.append(col3) 

print cols1
print cols2
print cols3
           

上面的代碼示例提出了讀取所有行的新寫法，提出了讀取某個cell和某一列的方式。

7、使用csv讀寫csv檔案示例代碼6

#encoding:utf-8

import csv

#第一種方法:使用reader函數,接收一個可疊代的對象(比如csv檔案),能傳回一個生成器,就可以從其中解析出csv的内容.eg:下面的代碼可以讀取csv的全部内容,以行為機關
with open('test.csv','rb') as csvFile:
	reader = csv.reader(csvFile)
	rows = [row for row in reader]
print rows 

print('===========') 

#提取其中某一列可以使用下面的方法
with open('test.csv','rb') as csvFile:
	reader = csv.reader(csvFile)
	column = [row[1] for row in reader]
print column


print('======第二種方法======')

#第二種方法:使用DictReader,和reader函數類似,接收可疊代的對象,能傳回一個可疊代的對象,能傳回一個生成器,但是傳回的每一個單元格都放在一個字典的值内,而這個字典的鍵則是這個單元格的标題.用下面的代碼可以看到DictReader的結構:

with open('test.csv','rb') as csvFile:
	reader = csv.DictReader(csvFile)
	column = [row for row in reader]
print(column) 

#使用CictReader讀取csv的某一列,這樣可以按照列的标題查詢

with open('test.csv','rb') as csvFile:
	reader = csv.DictReader(csvFile)
	column = [row['a_name'] for row in reader]
print column
           

8、使用csv讀寫csv檔案示例代碼7

#encoding:utf-8

import csv 

with open('test.csv','rb') as csvFile:
	reader = csv.reader(csvFile)
	for row in reader:
		print row

#向csv檔案中寫内容,這樣寫會把内容全部寫到一個cell中
with open('test1.csv','wb') as csvFile:
	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中
with open('test2.csv','wb') as csvFile:
	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的實作方式。