Python3:如何读取txt.bz2文件

时间:2017-02-26 10:27:05

标签: python-3.x bz2

有由bz2文件压缩的​​文本图块。 文本文件中的数据如下所示。

   1  x3, x32, f5

   0  f4, g6, h7, j9

   .............

我知道如何通过以下代码加载文本文件

 rf = open('small.txt', 'r')
    lines = rf.readlines()
    lst_text = []
    lst_label = []
    for line in lines:
        line = line.rstrip('\n')
        label, text = line.split('\t')
        text_words = text.split(',')
        lst_text.append(text_words)
        lst_label.append(int(label))

但是在将txt压缩为small.txt.bz2文件之后。 我想使用以下数据来读取bz2文件,但是有错误。

import bz2

bz_file = bz2.BZ2File("small.txt.bz2")
lines = bz_file.readlines()
for line in lines:
    line = line.rstrip('\n')
    label, text = line.split('\t')
    text_words = text.split(',')
    print(label)

错误:

      line = line.rstrip('\n')
TypeError: a bytes-like object is required, not 'str'

你能给我提示如何处理它,代码是最好的。谢谢!

1 个答案:

答案 0 :(得分:3)

您收到此错误,因为BZ2file对象在二进制模式下打开文件。因此,line bytes 对象,而不是字符串。您可以使用line = line.rstrip(b'\n')来解决这个问题。但结果行仍然是一个字节对象。

但您应该在文本模式下使用bz2.open

with bz2.open("small.txt.bz2", "rt") as bz_file:
    for line in bz_file:
        label, text = line.rstrip('\n').split('\t')
        text_words = text.split(',')
        print(label)