python - 使用 python-3.x 从 zip 存档中读取 CSV 文件

Question

我有一个包含多个 csv 文件的压缩存档。

例如，假设 myarchive.zip 包含 myfile1.csv、myfile2.csv、myfile3.csv

在 python 2.7 中，我能够使用 pandas 迭代加载所有 myfiles

import pandas as pd
import zipfile

with zipfile.ZipFile(myarchive.zip, 'r') as zippedyear:
 for filename in ['myfile1.csv', 'myfile2.csv', 'myfile3.csv']:
     mydf = pd.read_csv(zippedyear.open(filename))

现在用 Python 3 做同样的事情会抛出错误

ParserError: iterator should return strings, not bytes (did you open the file in text mode?)

我在这里不知所措。知道是什么问题吗？谢谢!

Answer 1

确实很奇怪，因为您可以指定的唯一模式是 r/w(字符模式)。

这里有一个解决方法；使用 file.read 读取文件，将数据加载到 StringIO 缓冲区，并将其传递给 read_csv。

from io import StringIO

with zipfile.ZipFile(myarchive.zip, 'r') as zippedyear:
    for filename in ['myfile1.csv', 'myfile2.csv', 'myfile3.csv']:
         with zippedyear.open(filename) as f:
             mydf = pd.read_csv(io.StringIO(f.read()))