解决python3的pickle.load错误：a bytes-like object is required, not 'str'

junlee87

于 2017-12-12 12:38:40 发布

阅读量1.9w

点赞数 21

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.csdn.net/junlee87/article/details/78780831

python 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种解决Python2环境下序列化的pickle文件在Python3中读取时出现的TypeError的方法。通过创建一个名为StrToBytes的类，可以将文件对象的read方法返回的str类型转换为bytes类型，从而实现pickle.load的正常工作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近在python3下使用pickle.load时出现了错误。

import pickle

with open('final_project_dataset.pkl', 'r') as data_file:
    data_dict = pickle.load(data_file)

错误信息如下：

    data_dict = pickle.load(data_file)

TypeError: a bytes-like object is required, not 'str'

经过几番查找，发现是Python3和Python2的字符串兼容问题，因为数据文件是在Python2下序列化的，所以使用Python3读取时，需要将‘str’转化为'bytes'。

class StrToBytes:
    def __init__(self, fileobj):
        self.fileobj = fileobj
    def read(self, size):
        return self.fileobj.read(size).encode()
    def readline(self, size=-1):
        return self.fileobj.readline(size).encode()

with open('final_project_dataset.pkl', 'r') as data_file:
    data_dict = pickle.load(StrToBytes(data_file))

经过这样一个转化后，就可以正确读取数据了。