如何读写csv数据

实际案例

http://table.finance.yahoo.com/table.csv?s=000001.sz(该网址已失效)
我们可以通过雅虎网址获取中国股市(深市)数据集,它以csv数据格式存储:

Date,Open,High,Low,Close,Volume,Adj Close
2016-06-30,8.69,8.74,8.66,8.70,36220400,8.70
2016-06-29,8.63,8.69,8.62,8.69,36961100,8.69
2016-06-28,8.58,8.64,8.56,8.63,33651900,8.63
......

请将平安银行这支股票,在2016年中成交量超过50000000的记录存储到另一个csv文件中。

解决方案:使用标准库中的csv模块,可以使用其中reader和writer完成csv文件读写。

基本代码为:

# -*- coding: utf-8 -*-

# -*- coding: utf-8 -*-

import csv
from urllib import urlretrieve

urlretrieve('http://table.finance.yahoo.com/table.csv?s=000001.sz', 'pingan.csv')

'''
# 读csv文件
rf = open('xxx.csv', 'rb')

reader = csv.reader(rf)
for row in reader:
    print row

# 写csv文件
wf = open('xxx.csv', 'wb')

writer = csv.writer(wf)
# 写入一个数据,数据可为列表
writer.writerow([])
'''

with open('pingan.csv', 'rb') as rf:

    reader = csv.reader(rf)
    with open('pingantest.csv', 'wb') as wf:

        writer = csv.writer(wf)
        headers = reader.next()
        writer.writerow(headers)
        for row in reader:

            if row[0] < '2016-01-01':

                break
            if int(row[5]) >= 50000000:

                writer.writerow(row)

print 'end'

注:由于教程录制于2016年,本人无法获取相关数据。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容