這篇文章主要介紹了使用Python對Csv檔案操作實例程式碼,非常具有實用價值,需要的朋友可以參考下
csv是Comma-Separated Values的縮寫,是用文字檔案形式儲存的表格數據,例如如下的表格:
#就可以儲存為csv文件,文件內容是:
No.,Name,Age,Score 1,mayi,18,99 2,jack,21,89 3,tom,25,95 4,rain,19,80
假設上述csv檔案儲存為"test.csv"
1.讀取檔案
如何用Python像操作Excel一樣擷取其中的一列,即一個字段,利用Python自帶的csv模組,有兩種方法可以實現:
第一種方法使用reader函數,接收一個可迭代的物件 (例如csv檔),能回傳一個產生器,就可以從其中解析出csv的內容:例如下面的程式碼可以讀取csv的全部內容,以行為單位:
#!/usr/bin/python3 # -*- conding:utf-8 -*- author = 'mayi' import csv #读 with open("test.csv", "r", encoding = "utf-8") as f: reader = csv.reader(f) rows = [row for row in reader] print(rows)
得到:
[['No.', 'Name', 'Age', 'Score'], ['1', 'mayi', '18', '99'], ['2', 'jack', '21', '89'], ['3', 'tom', '25', '95'], ['4', 'rain', '19', '80']]
要提取其中某一列,可以用下面的程式碼:
#!/usr/bin/python3 # -*- conding:utf-8 -*- author = 'mayi' import csv #读取第二列的内容 with open("test.csv", "r", encoding = "utf-8") as f: reader = csv.reader(f) column = [row[1] for row in reader] print(column)
得到:
['Name', 'mayi', 'jack', 'tom', 'rain']
注意從csv讀出的都是str類型。這種方法要事先知道列的序號,例如Name在第2列,而不能根據'Name'這個標題查詢#。這時可以採用第二種方法:
第二種方法是使用DictReader,和reader函數類似,接收一個可迭代的對象,能傳回一個生成器,但是返回的每一個單元格都放在一個字典的值內,而這個字典的鍵則是這個單元格的標題(即列頭)。用下面的程式碼可以看到DictReader的結構:
# -*- conding:utf-8 -*- author = 'mayi' import csv #读 with open("test.csv", "r", encoding = "utf-8") as f: reader = csv.DictReader(f) column = [row for row in reader] print(column)
得到:
[{'No.': '1', 'Age': '18', 'Score': '99', 'Name': 'mayi'}, {'No.': '2', 'Age': '21', 'Score': '89', 'Name': 'jack'}, {'No.': '3', 'Age': '25', 'Score': '95', 'Name': 'tom'}, {'No.': '4', 'Age': '19', 'Score': '80', 'Name': 'rain'}]
如果我們想用DictReader讀取csv的某一列,就可以用列的標題查詢:
#!/usr/bin/python3 # -*- conding:utf-8 -*- author = 'mayi' import csv #读取Name列的内容 with open("test.csv", "r", encoding = "utf-8") as f: reader = csv.DictReader(f) column = [row['Name'] for row in reader] print(column)
得到:
['mayi', 'jack', 'tom', 'rain']
2.寫檔案
#讀檔案時,我們把csv檔案讀入清單中,寫檔案時會把清單中的元素寫入到csv檔案中。
#!/usr/bin/python3 # -*- conding:utf-8 -*- author = 'mayi' import csv #写:追加 row = ['5', 'hanmeimei', '23', '81'] out = open("test.csv", "a", newline = "") csv_writer = csv.writer(out, dialect = "excel") csv_writer.writerow(row)
得到:
#以上是Python如何對Csv檔案操作的實例程式碼分享(圖)的詳細內容。更多資訊請關注PHP中文網其他相關文章!