Python 讀取csv的某行

2022-07-05 07:54:09 字數 4067 閱讀 4636

站長用python寫了乙個可以提取csv任一列的**,歡迎使用。github鏈結

如果我們想用dictreader讀取csv的某一列,就可以用列的標題查詢:

import csv

with open('a.csv','rb') as csvfile:

reader = csv.dictreader(csvfile)

for row in reader:

if row['name']=='ben':

print row就得到:

可見,dictreader很適合讀取csv的的行(記錄)。

csv (comma separated values),即逗號分隔值(也稱字元分隔值,因為分隔符可以不是逗號),是一種常用的文字

格式,用以儲存**資料,包括數字或者字元。很多程式在處理資料時都會碰到csv這種格式的檔案,它的使用是比

較廣泛的(kaggle上一些題目提供的資料就是csv格式),csv雖然使用廣泛,但卻沒有通用的標準,所以在處理csv

格式時常常會碰到麻煩,幸好python內建了csv模組。下面簡單介紹csv模組中最常用的一些函式。

引數說明:

csvfile,必須是支援迭代(iterator)的物件,可以是檔案(file)物件或者列表(list)物件,如果是檔案對

象,開啟時需要加"b"標誌引數。

dialect,編碼風格,預設為excel的風格,也就是用逗號(,)分隔,dialect方式也支援自定義,通過呼叫register_dialect方法來註冊,下文會提到。
fmtparam,格式化引數,用來覆蓋之前dialect物件指定的編碼風格。
[python]view plain

copy

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)只傳入了第乙個引數,另外兩個引數採用預設值,即以excel風格讀入。reader()返回乙個

reader物件lines,lines是乙個list,當呼叫它的方法lines.next()時,會返回乙個string。上面程式的效果是將csv

檔案中的文字按行列印,每一行的元素都是以逗號分隔符','分隔得來。

在我的test.csv檔案中,儲存的資料如圖:

程式輸出:

['1', '2']

['3', 'a']

['4', 'b']

引數的意義同上,這裡不贅述,直接上例程:

[python]view plain

copy

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()函式開啟當前路徑下的名字為't.csv'的檔案,如果不存在這個檔案,則建立它,返回myfile檔案物件。

csv.writer(myfile)返回writer物件mywriter。

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

注意:如果檔案't.csv'事先存在,呼叫writer函式會先清空原檔案中的文字,再執行writerow/writerows方法。

補充:除了writerow、writerows,writer物件還提供了其他一些方法:writeheader、dialect

這個函式是用來自定義dialect的。

引數說明:

name,你所自定義的dialect的名字,比如預設的是'excel',你可以定義成'mydialect'

[dialect, ]**fmtparams,dialect格式引數,有delimiter(分隔符,預設的就是逗號)、quotechar、

quoting等等,可以參考dialects and formatting parameters

[python]view plain

copy

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

上面一行程式自定義了乙個命名為mydialect的dialect,引數只設定了delimiter和quoting這兩個,其他的仍然採用

預設值,其中以'|'為分隔符。接下來我們就可以像使用'excel'一樣來使用'mydialect'了。我們來看看效果:

在我test.csv中儲存如下資料:

以'mydialect'風格列印:

[python]view plain

copy

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']

可以看到,現在是以'|'為分隔符,1和2合成了乙個字串(因為1和2之間的分隔符是逗號,而mydialect風格的分隔

符是'|'),3單獨乙個字串。

對於writer()函式,同樣可以傳入mydialect作為引數,這裡不贅述。

這個函式用於登出自定義的dialect

此外,csv模組還提供get_dialect(name)、list_dialects()、field_size_limit([new_limit])等函式,這些都比較

簡單,可以自己試試。比如list_dialects()函式會列出當前csv模組裡所有的dialect:

[python]view plain

copy

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的值。

先寫到這,其他的以後用到再更新。

Python 讀取csv的某行

站長用python寫了乙個可以提取csv任一列的 歡迎使用。github鏈結 就可以儲存為csv檔案,檔案內容是 假設上述csv檔案儲存為 a.csv 如何用python像操作excel一樣提取其中的一行,也就是一條記錄,利用python自帶的 csv模組,有兩種方法可以實現 第一種方法使用read...

Python 讀取csv的某行 兩種方法

站長用python寫了乙個可以提取csv任一列的 歡迎使用。github鏈結 就可以儲存為csv檔案,檔案內容是 假設上述csv檔案儲存為 a.csv 如何用python像操作excel一樣提取其中的一行,也就是一條記錄,利用python自帶的 csv模組,有兩種方法可以實現 第一種方法使用read...

c 讀取csv到陣列 python讀取csv檔案

python讀取csv檔案 csv檔案,用記事本編輯,然後命名為test1.csv 1,2,3,4 5,6,7,8 9,10,11,12 用excel開啟,就是這樣的 使用python讀取檔案 按行讀取。然後把讀到的東西賦值給python程式中的變數。這個變數,可以是乙個列表 在這裡,是乙個2維的列...