python讀寫csv文件方法詳細(xì)總結(jié)

sudo pip install pandas

sudo pip3 install pandas

#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環(huán)境下運(yùn)行
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")

#encoding:utf-8
import csv
 
#下面這種加了encoding的寫法需要在python3環(huán)境下運(yùn)行
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:
	#輸出的每一行是一個(gè)list,list中的每一個(gè)元素轉(zhuǎn)換成了string類型
	print(row)
		csv_writer.writerow(row)

#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文件
#設(shè)置newline,否則兩行之間會(huì)空一行
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()

#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存儲(chǔ)為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

#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])#打印一行中的某個(gè)cell
		print(row[0],row[1])#打印一行中的某個(gè)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

#encoding:utf-8
import csv
 
#第一種方法:使用reader函數(shù),接收一個(gè)可迭代的對(duì)象(比如csv文件),能返回一個(gè)生成器,就可以從其中解析出csv的內(nèi)容.eg:下面的代碼可以讀取csv的全部?jī)?nèi)容,以行為單位
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函數(shù)類似,接收可迭代的對(duì)象,能返回一個(gè)可迭代的對(duì)象,能返回一個(gè)生成器,但是返回的每一個(gè)單元格都放在一個(gè)字典的值內(nèi),而這個(gè)字典的鍵則是這個(gè)單元格的標(biāo)題.用下面的代碼可以看到DictReader的結(jié)構(gòu):
with open('test.csv','rb') as csvFile:
	reader = csv.DictReader(csvFile)
	column = [row for row in reader]
print(column)
 
#使用CictReader讀取csv的某一列,這樣可以按照列的標(biāo)題查詢
with open('test.csv','rb') as csvFile:
	reader = csv.DictReader(csvFile)
	column = [row['a_name'] for row in reader]
print column

#encoding:utf-8
import csv
 
with open('test.csv','rb') as csvFile:
	reader = csv.reader(csvFile)
	for row in reader:
		print row
 
#向csv文件中寫內(nèi)容,這樣寫會(huì)把內(nèi)容全部寫到一個(gè)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文件中寫內(nèi)容,下面的寫法最終可以把每一列的內(nèi)容放在一個(gè)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'])