python讀寫csv檔案方法總結
python提供了大量的庫,可以非常方便的進行各種操作,現在把python中實現讀寫csv檔案的方法使用程式的方式呈現出來。
1、使用csv讀寫csv檔案方法總結
reader()函式是一個閱讀器把閱讀的CSV檔案每一行以一個列表表示出來以至於你可以用for迴圈來遍歷他
讀檔案的時候,開啟檔案,呼叫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
上面的程式碼提出了讀取整行和整列的新方式。另外提出了一種使用DictReader的方式按照列名讀取的方式。
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的實現方式。
5、使用csv讀寫csv檔案示例程式碼4
作者:jp_666
來源:CSDN
原文:https://blog.csdn.net/jp_666/article/details/78703068
版權宣告:本文為博主原創文章,轉載請附上博文連結!