6-1 如何读写csv数据

实际案例：

• http://table.finance.yahoo.com/table.csv?s=000001.sz我们可以通过雅虎网站获取中国股市数据集，它以csv数据格式存储:
• Date,Open,High,Low,Close,Volume,AdjClose
  2016-06-30,8.69,8.74,8.66,8.70,36220400,8.70
  ....
  *请将平安银行这支股票，在2016年中成交量超过500000000的记录存储到另一个csv文件中.

解决方案：

• 使用标准库中的csv模块，可以使用其中reader和writer完成csv文件读写.

from urllib.request import urlretrieve
urlretrieve(r'https://table.finance.yahoo.com/table.csv?s=000001.sz',r'pingan.csv')

from csv
rf = open('pingan.csv','rb')
reader = csv.reader(rf)

for row in reader:
  print(row)

rf.seek(0)

wf = open('pingan_copy.csv','wb')
writer = csv.writer(wf)

writer.writerow(reader.next())
for row in reader:
  if row[0] < '2016-01-01':
    break
  if row[5] > 500000000:
    writer.writerow(row)

wf.flush()
wf.close()
rf.close()