Python中csv模塊怎么用

這篇文章主要介紹了Python中csv模塊怎么用，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

1、csv簡(jiǎn)介

CSV (Comma Separated Values) ，即逗號(hào)分隔值（也稱(chēng)字符分隔值，因?yàn)榉指舴梢圆皇嵌禾?hào)），是一種常用的文本

格式，用以存儲(chǔ)表格數(shù)據(jù)，包括數(shù)字或者字符。很多程序在處理數(shù)據(jù)時(shí)都會(huì)碰到csv這種格式的文件，它的使用是比

較廣泛的（Kaggle上一些題目提供的數(shù)據(jù)就是csv格式），csv雖然使用廣泛，但卻沒(méi)有通用的標(biāo)準(zhǔn)，所以在處理csv

格式時(shí)常常會(huì)碰到麻煩，幸好python內(nèi)置了csv模塊。下面簡(jiǎn)單介紹csv模塊中最常用的一些函數(shù)。

2、csv模塊中的函數(shù)

reader(csvfile, dialect='excel', **fmtparams)

參數(shù)說(shuō)明：

csvfile，必須是支持迭代(Iterator)的對(duì)象，可以是文件(file)對(duì)象或者列表(list)對(duì)象，如果是文件對(duì)
象，打開(kāi)時(shí)需要加"b"標(biāo)志參數(shù)。

dialect，編碼風(fēng)格，默認(rèn)為excel的風(fēng)格，也就是用逗號(hào)（,）分隔，dialect 方式也支持自定義，通過(guò)調(diào)用register_dialect方法來(lái)注冊(cè)，下文會(huì)提到。

fmtparam，格式化參數(shù)，用來(lái)覆蓋之前dialect對(duì)象指定的編碼風(fēng)格。

import csv
with open('test.csv','rb') as myFile:
    lines=csv.reader(myFile)
    for line in lines:
        print line

'test.csv'是文件名，‘rb'中的r表示“讀”模式，因?yàn)槭俏募?duì)象，所以加‘b'。open()返回了一個(gè)文件對(duì)象

myFile，reader(myFile)只傳入了第一個(gè)參數(shù)，另外兩個(gè)參數(shù)采用缺省值，即以excel風(fēng)格讀入。reader()返回一個(gè)

reader對(duì)象lines,lines是一個(gè)list，當(dāng)調(diào)用它的方法lines.next()時(shí)，會(huì)返回一個(gè)string。上面程序的效果是將csv

文件中的文本按行打印，每一行的元素都是以逗號(hào)分隔符','分隔得來(lái)。

在我的test.csv文件中，存儲(chǔ)的數(shù)據(jù)如圖：

程序輸出：

['1', '2']
['3', 'a']
['4', 'b']

補(bǔ)充：reader對(duì)象還提供一些方法：line_num、dialect、next()

writer(csvfile, dialect='excel', **fmtparams)

參數(shù)的意義同上，這里不贅述，直接上例程：

with open('t.csv','wb') as myFile:    
    myWriter=csv.writer(myFile)
    myWriter.writerow([7,'g'])
    myWriter.writerow([8,'h'])
    myList=[[1,2,3],[4,5,6]]
    myWriter.writerows(myList)

'w'表示寫(xiě)模式。

首先open()函數(shù)打開(kāi)當(dāng)前路徑下的名字為't.csv'的文件，如果不存在這個(gè)文件，則創(chuàng)建它，返回myFile文件對(duì)象。

csv.writer(myFile)返回writer對(duì)象myWriter。

writerow()方法是一行一行寫(xiě)入，writerows方法是一次寫(xiě)入多行。

注意：如果文件't.csv'事先存在，調(diào)用writer函數(shù)會(huì)先清空原文件中的文本，再執(zhí)行writerow/writerows方法。

補(bǔ)充：除了writerow、writerows，writer對(duì)象還提供了其他一些方法：writeheader、dialect

register_dialect(name, [dialect, ]**fmtparams)

這個(gè)函數(shù)是用來(lái)自定義dialect的。

參數(shù)說(shuō)明：

name,你所自定義的dialect的名字，比如默認(rèn)的是'excel'，你可以定義成'mydialect'

[dialect, ]**fmtparams，dialect格式參數(shù)，有delimiter（分隔符，默認(rèn)的就是逗號(hào)）、quotechar、

quoting等等，可以參考Dialects and Formatting Parameters

csv.register_dialect('mydialect',delimiter='|', quoting=csv.QUOTE_ALL)

上面一行程序自定義了一個(gè)命名為mydialect的dialect，參數(shù)只設(shè)置了delimiter和quoting這兩個(gè)，其他的仍然采用

默認(rèn)值，其中以'|'為分隔符。接下來(lái)我們就可以像使用'excel'一樣來(lái)使用'mydialect'了。我們來(lái)看看效果：

在我test.csv中存儲(chǔ)如下數(shù)據(jù)：

以'mydialect'風(fēng)格打?。?/p>

with open('test.csv','rb') as myFile:
    lines=csv.reader(myFile,'mydialect')
    print lines.line_num
    for line in lines:
        print line

輸出：

['1,2', '3']
['4,5', '6']

可以看到，現(xiàn)在是以'|'為分隔符，1和2合成了一個(gè)字符串（因?yàn)?和2之間的分隔符是逗號(hào)，而mydialect風(fēng)格的分隔

符是'|'），3單獨(dú)一個(gè)字符串。

對(duì)于writer()函數(shù)，同樣可以傳入mydialect作為參數(shù)，這里不贅述。

unregister_dialect(name)

這個(gè)函數(shù)用于注銷(xiāo)自定義的dialect

此外，csv模塊還提供get_dialect(name)、list_dialects()、field_size_limit([new_limit])等函數(shù)，這些都比較

簡(jiǎn)單，可以自己試試。比如list_dialects()函數(shù)會(huì)列出當(dāng)前csv模塊里所有的dialect：

print csv.list_dialects()

輸出：

['excel-tab', 'excel', 'mydialect']

'mydialect'是自定義的，'excel-tab', 'excel'都是自帶的dialect，其中'excel-tab'跟'excel'差不多，

只不過(guò)它以tab為分隔符。

csv模塊還定義了

一些類(lèi)：DictReader、DictWriter、Dialect等，DictReader和DictWriter類(lèi)似于reader和writer。

一些常量：QUOTE_ALL、QUOTE_MINIMAL、.QUOTE_NONNUMERIC等，這些常量可以作為Dialects and Formatting Parameters的值。

先寫(xiě)到這，其他的以后用到再更新。

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“Python中csv模塊怎么用”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來(lái)學(xué)習(xí)!