在python里面,讀取或寫入csv文件時,首先要import csv這個庫,然后利用這個庫提供的方法進行對文件的讀寫。
0x01:獲取每一行
讀取csv文件,用的是csv.reader()這個方法。返回結果是一個_csv.reader的對象,我們可以對這個對象進行遍歷,輸出每一行,某一行,或某一列。代碼如下:
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 print(type(reader)) 5 6 for row in reader: 7 print(row)
以列表的形式輸出每一行,如下:
0x02:獲取某一行
如果你只想要其中的一行,可以先對reader進行一個類型轉換,用list函數把它轉換成列表,然后對對列表進行取元素,就可以獲取到某一行的內容了,如下:
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 result = list(reader) 5 print(result[1])
這里,我們獲取的是第二行的內容,與上面的第二行的內容相同。
0x03:獲取某一列
如果想要獲取到其中一列的內容,可以在第一種情況下,對輸出加一個下標,這樣輸出的就是某一列的內容的。
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 for i in reader: 5 print(i[0])
這里,獲取第一列的內容,與第一次輸出結果相同