Question

我正在尝试从问题Saving interactive Matplotlib figures中重新创建简单的泡菜图示例，这也来自Saving Matplotlib Figures Using Pickle。但是，当我运行给定的代码时，数字看起来很好，但是当我尝试加载酸洗的数字时，我得到一个错误。我使用Canopy Enthought（v1.6.2.3262）运行它，在Python 2.7.3-1上使用Matplotlib 1.5.1-1和Numpy 1.9.2-3。泡菜代码是：`

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中的转储（我认为是pickle加载）应该使用而不是open文件，因此我在代码中尝试了两种组合。

我不确定错误告诉我的是什么，所以我还没有能够进一步了解这一点，任何有助于理解错误或解决问题的帮助。

Answer 1

copy_reg的错误是由代码中的写入格式引起的，正确的代码应该包含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错误的解决方案而确定的。

Pickle数字来自matplotlib

1 个答案: