有由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'
你能给我提示如何处理它,代码是最好的。谢谢!
答案 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)