python如何读写csv数据

yipeiwu_com5年前Python基础

本文实例为大家分享了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()

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

Python的Tornado框架的异步任务与AsyncHTTPClient

高性能服务器Tornado Python的web框架名目繁多,各有千秋。正如光荣属于希腊,伟大属于罗马。Python的优雅结合WSGI的设计,让web框架接口实现千秋一统。WSGI 把应...

利用python实现周期财务统计可视化

利用python实现周期财务统计可视化

正文之前 上午给爸爸打了个电话庆祝他50岁生日,在此之前搞了个大扫除,看了会知乎,到实验室已经十一点多了。约喜欢的妹子吃饭失败,以至于工作积极性收到了打击,所以就写个程序来统计下开学十一...

Python内置函数 next的具体使用方法

Python 3中的File对象不支持next()方法。 Python 3有一个内置函数next(),它通过调用其next ()方法从迭代器中检索下一个项目。 如果给定了默认值,则在迭代...

Python实现的读写json文件功能示例

本文实例讲述了Python实现的读写json文件功能。分享给大家供大家参考,具体如下: 相比java,python对json文件的处理就简单很多。java操作json文件的话需要引用ja...

python 从文件夹抽取图片另存的方法

有一个比较大的数据集需要自己处理,在分出训练集和测试集时,如果靠手动实在太麻烦,于是自己写了一段代码。(其实就是在某一路径下的子文件夹里取出符合要求的图片,放到另一个路径的对应文件夹中)...