本文實(shí)例為大家分享了python讀寫csv數(shù)據(jù)的具體代碼,供大家參考,具體內(nèi)容如下
案例:
通過股票網(wǎng)站,我們獲取了中國股市數(shù)據(jù)集,它以csv數(shù)據(jù)格式存儲(chǔ)
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
......
需求:請(qǐng)將2016年成交量超過500000記錄存儲(chǔ)到另外一個(gè)csv文件中
什么是csv類型數(shù)據(jù)?
類似表單,第一行數(shù)據(jù)是字段名字,單獨(dú)數(shù)據(jù)以逗號(hào)隔開
如何做?
獲取股票信息的csv類型數(shù)據(jù)
以二進(jìn)制讀的形式打開csv類型數(shù)據(jù),csv.reader(文件對(duì)象)返回一個(gè)對(duì)象,通過對(duì)象操作
以二進(jìn)制寫的形式打開新的csv數(shù)據(jù),csv.writerrow(文件對(duì)象)返回一個(gè)對(duì)象,通過對(duì)象操作
循環(huán)讀取csv中數(shù)據(jù),通過if判斷是否滿足需求,滿足的寫入到新的csv文件中
#!/usr/bin/python3 import csv def r_w_csv(): with open('pingan.csv', 'rb') as rf: # 打開原文件讀取進(jìn)來,csv.read(rf)方式 reader = csv.read(rf) with open('pingan_copy.csv', 'wb') as wf: # 打開另外一個(gè)文件,csv.writer(wf)方式,寫入csv數(shù)據(jù)頭部 writer = csv.writer(wf) headers = reader.next() writer.writerrow(headers) # 給索引命名,增加可讀性 data, amount = 0, 5 # 寫入時(shí)間2016年,營業(yè)額大于等于50000000的數(shù)據(jù) 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()以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持武林站長站。
新聞熱點(diǎn)
疑難解答
圖片精選