脚本宝典收集整理的这篇文章主要介绍了

直接从压缩包中读取,遍历CSV文件

脚本宝典小编觉得挺不错的,现在分享给大家,也给大家做个参考,希望能帮助你少写一行代码,多一份安全和惬意。

最近这段时间遇到了一个问题,直接打开压缩包文件后,用csv.reader读后,用for in 进行遍历时,会提示:iterator should return strings not bytes (did you open the file in text mode ?)。出问题的代码是这样的:

import io
import zipfile
import csv

with zipfile.ZipFile('./12.zip') as zf:
    name = zf.namelist()[0]
    csvfile = zf.open(name)
    for x,y in csv.reader(csvfile):
        print(x,'的英文缩写是:',y)

分析原因后,主要是因为 ZipFile 读进来的内容是bytes类型,csv.reader读进去的类型也是bytes,造成无法遍历。
想到的第一个办法是用decode()进行转换,但csv.reader的参数应该是一个文件流,而非字符串,需要用io.StringIO()进行封装。 有效的代码如下:

with zipfile.ZipFile('./12.zip') as zf:
    name = zf.namelist()[0]
    csvfile = io.StringIO(zf.open(name).read().decode('gbk'))
    for x,y in csv.reader(csvfile):
        print(x,'的英文缩写是:',y)

这样问题就得以解决。

打开硬盘中的文件多数是采用流的方式进行的,实际上在本例中,ZipFile实现了从硬盘到内存的流操作,zf.open实现了从压缩文件到csv文件的流操作,io.StringIO封装了对字符串的流。

总结

以上是脚本宝典为你收集整理的

直接从压缩包中读取,遍历CSV文件

全部内容,希望文章能够帮你解决

直接从压缩包中读取,遍历CSV文件

所遇到的程序开发问题,欢迎加入QQ群277859234一起讨论学习。如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典网站推荐给程序员好友。 本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。

80%的人都看过