站長用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'[python]view plain[dialect, ]**fmtparams,dialect格式引數,有delimiter(分隔符,預設的就是逗號)、quotechar、
quoting等等,可以參考dialects and formatting parameters
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維的列...