1. 程式人生 > >Python中的CSV模組基本用法

Python中的CSV模組基本用法

CSV模組提供一種非常便捷的方式操作CSV檔案。本文只是展現最基本的csv檔案的讀寫操作,因為使用的是Python2.7 所以open接受引數少,整個csv讀寫例子如下

import csv
fname = 'testcsv.csv'
with open(fname,'wb') as csvfile: # 解決寫入空行問題 使用wb不會再每一行後面插入空行
    csvwriter = csv.writer(csvfile,delimiter=',')
    lst= [[1,2,3],[4,5,6]]
    for item in lst:
        csvwriter.writerow(item)
# 讀取操作
with open(fname,'r'
) as csvfile: rows = csv.reader(csvfile) for row in rows: print row print type(row) # 型別為一個list

執行結果

['1', '2', '3']
<type 'list'>
['4', '5', '6']
<type 'list'>

有時在python3 下需要將矩陣資料寫入CSV檔案,也會出現空行. 但是按照上述方式會報錯,需要資料為bytes 型別,為了解決這個問題,需要在open的時候指定一個newline=''引數,讓它為空,可以保證CSV寫入資料不會出現空行.

import csv
fname = 'testcsv.csv'
with open(fname,'w', newline='') as csvfile: # 解決寫入空行問題 使用wb不會再每一行後面插入空行
    csvwriter = csv.writer(csvfile)
    lst= [[1,2,3],[4,5,6]]
    csvwriter.writerows(lst)