Python中csv模塊如何使用

這篇文章主要介紹“Python中csv模塊如何使用”，在日常操作中，相信很多人在Python中csv模塊如何使用問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Python中csv模塊如何使用”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

1、csv簡介

CSV (Comma Separated Values) ，即逗號分隔值（也稱字符分隔值，因為分隔符可以不是逗號），是一種常用的文本

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

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

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

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

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

參數(shù)說明：

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

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

fmtparam，格式化參數(shù)，用來覆蓋之前dialect對象指定的編碼風(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表示“讀”模式，因為是文件對象，所以加‘b"。open()返回了一個文件對象

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

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

文件中的文本按行打印，每一行的元素都是以逗號分隔符","分隔得來。

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

程序輸出：

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

補充：reader對象還提供一些方法：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"表示寫模式。

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

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

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

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

補充：除了writerow、writerows，writer對象還提供了其他一些方法：writeheader、dialect

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

這個函數(shù)是用來自定義dialect的。

參數(shù)說明：

name,你所自定義的dialect的名字，比如默認的是"excel"，你可以定義成"mydialect"

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

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

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

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

默認值，其中以"|"為分隔符。接下來我們就可以像使用"excel"一樣來使用"mydialect"了。我們來看看效果：

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

以"mydialect"風(fēng)格打印：

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合成了一個字符串（因為1和2之間的分隔符是逗號，而mydialect風(fēng)格的分隔

符是"|"），3單獨一個字符串。

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

unregister_dialect(name)

這個函數(shù)用于注銷自定義的dialect

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

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

print csv.list_dialects()

輸出：

["excel-tab", "excel", "mydialect"]

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

只不過它以tab為分隔符。

csv模塊還定義了

一些類：DictReader、DictWriter、Dialect等，DictReader和DictWriter類似于reader和writer。

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

到此，關(guān)于“Python中csv模塊如何使用”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>