python提供了大量的庫(kù)，可以非常方便的進(jìn)行各種操作，現(xiàn)在把python中實(shí)現(xiàn)讀寫(xiě)csv文件的方法使用程序的方式呈現(xiàn)出來(lái)。
在編寫(xiě)python程序的時(shí)候需要csv模塊或者pandas模塊，其中csv模塊使不需要重新下載安裝的，pandas模塊需要按照對(duì)應(yīng)的
python版本安裝。
在python2環(huán)境下安裝pandas的方式是：
sudo pip install pandas
在python3環(huán)境下安裝pandas的方式是：
sudo pip3 install pandas
1、使用csv讀寫(xiě)csv文件方法總結(jié)
讀文件的時(shí)候，打開(kāi)文件，調(diào)用csv.reader()讀取文件；對(duì)于讀取之后的文件的內(nèi)容，要把這些內(nèi)容輸入到另一個(gè)文件中保存，可以通過(guò)遍歷讀取的文件的每一行，然后使用csv_write.writerow()的方式寫(xiě)入到指定的文件。
2、使用csv讀寫(xiě)csv文件示例代碼
#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")
上面的代碼是傳統(tǒng)的打開(kāi)文件---讀取文件---打開(kāi)文件---關(guān)閉文件的方式進(jìn)行文件讀取和寫(xiě)入。在讀取和寫(xiě)入的過(guò)程中分別調(diào)用了csv庫(kù)的reader()和writer()
3、使用csv讀寫(xiě)csv文件示例代碼
#encoding:utf-8
import csv
#下面這種加了encoding的寫(xiě)法需要在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)
示例代碼2的讀寫(xiě)方式與示例代碼1的方式基本相同，不同的是在示例代碼2中打開(kāi)文件的時(shí)候指定了打開(kāi)時(shí)的編碼方式，對(duì)于輸出文件的打開(kāi)方式也與示例代碼1的方式有少量的差別，把寫(xiě)入方式右'a'改成了'w
4、使用csv讀寫(xiě)csv文件示例代碼
#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()
#從列表寫(xiě)入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()
#從字典寫(xiě)入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()
在上面的代碼中提出了兩種不同的打開(kāi)csv文件和寫(xiě)入csv文件的方式。其中第一種打開(kāi)方式與前面兩種非常相似，不同之處是在這種方式中，打開(kāi)文件的時(shí)候加入了讀取方式，而前面兩個(gè)示例代碼沒(méi)有加。另一種讀取方式是采用python中獨(dú)有的命名方式對(duì)打開(kāi)文件命名，其他部分相似。在這個(gè)示例中，對(duì)列表和字典寫(xiě)入csv文件做了區(qū)分，分別提供了列表和字典寫(xiě)入csv文件的示例代碼。
5、使用csv讀寫(xiě)csv文件示例代碼
#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包,一行一行寫(xiě)入
import csv
#python2可以用file替代open
with open('test.csv','w') as csvFile:
writer = csv.writer(csvFile)
#先寫(xiě)columns_name
writer.writerow(["index","a_name","b_name"])
#寫(xiě)入多行用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
這個(gè)示例代碼提出了寫(xiě)入一行和寫(xiě)入多行的實(shí)現(xiàn)方式。
6、使用csv讀寫(xiě)csv文件示例代碼
#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
上面的代碼示例提出了讀取所有行的新寫(xiě)法，提出了讀取某個(gè)cell和某一列的方式。
7、使用csv讀寫(xiě)csv文件示例代碼
#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
上面的代碼提出了讀取整行和整列的新方式。另外提出了一種使用DictReader的方式按照列名讀取的方式。
8、使用csv讀寫(xiě)csv文件示例代碼
#encoding:utf-8
import csv
with open('test.csv','rb') as csvFile:
reader = csv.reader(csvFile)
for row in reader:
print row
#向csv文件中寫(xiě)內(nèi)容,這樣寫(xiě)會(huì)把內(nèi)容全部寫(xiě)到一個(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文件中寫(xiě)內(nèi)容,下面的寫(xiě)法最終可以把每一列的內(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'])
上面的代碼提出了可以把列表中的數(shù)據(jù)放入到不同cell的實(shí)現(xiàn)方式。
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號(hào)聯(lián)系： 360901061
您的支持是博主寫(xiě)作最大的動(dòng)力，如果您喜歡我的文章，感覺(jué)我的文章對(duì)您有幫助，請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點(diǎn)擊下面給點(diǎn)支持吧，站長(zhǎng)非常感激您！手機(jī)微信長(zhǎng)按不能支付解決辦法：請(qǐng)將微信支付二維碼保存到相冊(cè)，切換到微信，然后點(diǎn)擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元
