Python pickle模块踩坑

跟着机器学习实战写代码,决策树这里有一段是保存决策树,使用pickle模块保存,原书是基于2.7的,在3.6上有坑,在这里跟大家分享下。

'''

def storeTree(inputTree, filename):

import pickle

fw = open(filename, 'w')

pickle.dump(inputTree,fw)

fw.close()

def grabTree(filename):

import pickle

fr = open(filename, )

print(pickle.load(fr))

'''

这是原代码,跑完会告诉你2进制文件无法打开a bytes-like object is required, not 'str',第一步在上面储存的函数里将w改为wb,二进制写模式。这样改完不够还需要在下面读的文件里加上rb,已二进制文件读取。除此之外还不够,还有一个最隐蔽的,open的时候有4中模式,默认是第三模式,这里要想正确读出只能选择0模式,即ASCII模式。0:ASCII协议,所序列化的对象使用可打印的ASCII码表示;1:老式的二进制协议;2:2.3版本引入的新二进制协议,较以前的更高效。其中协议0和1兼容老版本的python。protocol默认值为3。

改完之后就可以用了,新代码在下面

'''

def storeTree(inputTree, filename):

import pickle

fw = open(filename, 'wb')

pickle.dump(inputTree,fw,0)

fw.close()

def grabTree(filename):

import pickle

fr = open(filename, 'rb')

print(pickle.load(fr))

'''

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1. Java基础部分 基础部分的顺序:基本语法,类相关的语法,内部类的语法,继承相关的语法,异常的语法,线程的语...
    子非鱼_t_阅读 31,765评论 18 399
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,937评论 18 139
  • http://python.jobbole.com/85231/ 关于专业技能写完项目接着写写一名3年工作经验的J...
    燕京博士阅读 7,628评论 1 118
  • 认识你时,我们都错过了彼此了青涩年华,我看不到你原来的自私、无情、冲动,你也不知道我原来的任性、无理、野蛮。 无意...
    shuishui阅读 227评论 0 0
  • 朴有天,我感觉你变了,不是我喜欢的朴有天了。虽然人都会变,可是真的很可怕。也可能,不是你变了,是我变了
    让我死在意外阅读 208评论 0 0