python如何讀寫csv資料,python讀寫csv資料
本文執行個體為大家分享了python讀寫csv資料的具體代碼,供大家參考,具體內容如下
案例:
通過股票網站,我們擷取了中國股市資料集,它以csv資料格式儲存
Data,Open,High,Low,Close,Volume,Adj Close
2016-06-28,8.63,8.47,8.66,8.70,500000,8.70
2016-06-28,8.63,8.47,8.66,8.70,500000,8.70
2016-06-28,8.63,8.47,8.66,8.70,500000,8.70
......
需求:請將2016年成交量超過500000記錄儲存到另外一個csv檔案中
什麼是csv類型資料?
類似表單,第一行資料是欄位名字,單獨資料以逗號隔開
如何做?
擷取股票資訊的csv類型資料
以二進位讀的形式開啟csv類型資料,csv.reader(檔案對象)返回一個對象,通過對象操作
以二進位寫的形式開啟新的csv資料,csv.writerrow(檔案對象)返回一個對象,通過對象操作
迴圈讀取csv中資料,通過if判斷是否滿足需求,滿足的寫入到新的csv檔案中
#!/usr/bin/python3 import csv def r_w_csv(): with open('pingan.csv', 'rb') as rf: # 開啟原檔案讀取進來,csv.read(rf)方式 reader = csv.read(rf) with open('pingan_copy.csv', 'wb') as wf: # 開啟另外一個檔案,csv.writer(wf)方式,寫入csv資料頭部 writer = csv.writer(wf) headers = reader.next() writer.writerrow(headers) # 給索引命名,增加可讀性 data, amount = 0, 5 # 寫入時間2016年,營業額大於等於50000000的資料 for row in reader: if row[data] < '2016-01-01': break if int(row[amount]) >= 50000000: writer.writerrow(row) if __name__ == '__main__': r_w_csv()
以上就是本文的全部內容,希望對大家的學習有所協助,也希望大家多多支援幫客之家。