python - 来自 matplotlib 的 pickle 数字

Question

我试图从问题中重新创建简单的 pickle 图示例:Saving interactive Matplotlib figures ,
这也来自 Saving Matplotlib Figures Using Pickle .但是，当我运行给定的代码时，数字似乎可以 pickle ，但是当我尝试加载 pickle 数字时出现错误。我正在使用 Canopy Enthought (v1.6.2.3262) 运行它，在 Python 2.7.3-1 上使用 Matplotlib 1.5.1-1 和 Numpy 1.9.2-3。
pickle 代码是:`

import numpy as np
import matplotlib.pyplot as plt
import pickle as pl

# Plot simple sinus function
fig_handle = plt.figure()
x = np.linspace(0,2*np.pi)
y = np.sin(x)
plt.plot(x,y)

# Save figure handle to disk
pl.dump(fig_handle,file('sinus.pickle','w'))`

加载图形的代码是:

import matplotlib.pyplot as plt
import pickle as pl
import numpy as np

# Load figure from disk and display
fig_handle = pl.load(open('sinus.pickle','rb'))
fig_handle.show()

我得到的错误是:

%run "Z:\EFNHigh_Res\show_picklefig.py"
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Z:\EFNHigh_Res\show_picklefig.py in <module>()
      4 
      5 #plot simple sinus function
----> 6 fig_handle = pl.load(open('Z:\EFNHigh_Res\sinus.pickle','rb'))
      7 fig_handle.show()

C:\Users\Tom\AppData\Local\Enthought\Canopy\App\appdata\canopy-1.6.2.3262.win-x86_64\lib\pickle.pyc in load(file)
   1376 
   1377 def load(file):
-> 1378     return Unpickler(file).load()
   1379 
   1380 def loads(str):

C:\Users\Tom\AppData\Local\Enthought\Canopy\App\appdata\canopy-1.6.2.3262.win-x86_64\lib\pickle.pyc in load(self)
    856             while 1:
    857                 key = read(1)
--> 858                 dispatch[key](self)
    859         except _Stop, stopinst:
    860             return stopinst.value

C:\Users\Tom\AppData\Local\Enthought\Canopy\App\appdata\canopy-1.6.2.3262.win-x86_64\lib\pickle.pyc in load_global(self)
   1088         module = self.readline()[:-1]
   1089         name = self.readline()[:-1]
-> 1090         klass = self.find_class(module, name)
   1091         self.append(klass)
   1092     dispatch[GLOBAL] = load_global

C:\Users\Tom\AppData\Local\Enthought\Canopy\App\appdata\canopy-1.6.2.3262.win-x86_64\lib\pickle.pyc in find_class(self, module, name)
   1122     def find_class(self, module, name):
   1123         # Subclasses may override this
-> 1124         __import__(module)
   1125         mod = sys.modules[module]
   1126         klass = getattr(mod, name)

ImportError: No module named copy_reg

我知道 Python 3 和 2 之间存在差异，在 Python 2 的转储中应该使用该文件而不是 open 文件(并且我假设是pickle 负载)，所以我在代码中尝试了这两种组合。

我不确定错误告诉我什么，所以我无法进一步了解这一点，对理解错误或解决问题的任何帮助表示赞赏。

Answer 1

copy_reg 的错误是由代码中的写入格式引起的，以pickle 数字，正确的代码应该在 write 语句中包含 wb 而不是 w，如下所示:

# Save figure handle to disk
import pickle
with open('sinus.pickle', 'wb') as f: # should be 'wb' rather than 'w'
    pickle.dump(fig_handle, f) 

这是基于 copy_reg 错误和另一个问题 ImportError: No module named copy_reg pickle 中提供的解决方案确定的关于 pickle 时的 copy_reg 错误。