python - 将二进制输入流转换为文本模式

Question

我正在尝试在 Python 3.2 中读取 bzip2 压缩的 CSV 文件。对于未压缩的 CSV 文件，此方法有效：

datafile = open('./file.csv', mode='rt')
data = csv.reader(datafile)
for e in data:    # works
    process(e)

问题是BZ2File它只支持创建二进制流，而在 Python 3 中，csv.reader只接受文本流。（gzip 和 zip 文件也会出现同样的问题。）

datafile = bz2.BZ2File('./file.csv.bz2', mode='r')
data = csv.reader(datafile)
for e in data:    # error
    process(e)

特别是，指示的行会引发异常_csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)。

我也试过data = csv.reader(codecs.EncodedFile(datafile, 'utf8'))了，但这并不能解决错误。

如何包装二进制输入流以便它可以在文本模式下使用？

score 5 · Accepted Answer

这对我有用：

import codecs, csv
f = codecs.open("file.csv", "r", "utf-8")
g = csv.reader(f)
for e in g:
    print(e)

在 BZ2 的情况下：

import codecs, csv, bz2
f = bz2.BZ2File("./file.csv.bz2", mode="r")
c = codecs.iterdecode(f, "utf-8")
g = csv.reader(c)
for e in g:
    print(e)

1 回答 1