python - 如何 pickle numpy 的 Inf 对象？

Question

当尝试 pickle numpy 中定义的对象 Inf 时(我认为)，转储正常但加载失败:

>>> cPickle.dump(Inf, file("c:/temp/a.pcl",'wb'))
>>> cPickle.load(file("c:/temp/a.pcl",'rb'))
Traceback (most recent call last):
  File "<pyshell#257>", line 1, in <module>
    cPickle.load(file("c:/temp/a.pcl",'rb'))
ValueError: could not convert string to float
>>> type(Inf)
<type 'float'>

这是为什么呢？而且 - 有没有办法解决这个问题？我想 pickle 其中包含 Inf 的东西 - 将其更改为其他东西会破坏程序的优雅......

谢谢

Answer 1

如果您指定的 pickle 协议(protocol)大于零，它将起作用。协议(protocol)通常指定为-1，表示使用最新最好的协议(protocol):

>>> cPickle.dump(Inf, file("c:/temp/a.pcl",'wb'), -1)
>>> cPickle.load(file("c:/temp/a.pcl",'rb'))
1.#INF                   -- may be platform dependent what prints here.