将python matplotlib图形另存为pickle

时间:2018-07-10 11:51:38

标签: python matplotlib pickle

我想将matplotlib图形保存为泡菜,以使我的团队能够轻松地加载它们并调查异常事件。使用简单的图像格式会失去放大和研究图形的能力。因此,我试图腌制我的身材:

fig, axarr = plt.subplots(2, sharex='all') # creating the figure
... # plotting things
...
pickle.dump(fig, open('myfigfile.p'), 'wb'))

然后我加载它。看起来不错。首先。

fig = pickle.load(open('myfigfile.p', 'rb'))
plt.show()

但是随后我发现sharex无法正常工作。当我放大一个子图时,其他子图保持不变。在原始图上,这很好用,并且可以在两个x轴上放大,但是在给该图腌制之后,这不再起作用。 如何保存绘图,以便在加载后继续共享x? 或者,从咸菜中加载图形后,如何恢复share-x?

谢谢!

1 个答案:

答案 0 :(得分:4)

在当前的matplotlib开发版本中,由于fix provided,这应该可以工作。

使用matplotlib的当前发行版本,需要重新建立共享,例如喜欢

import matplotlib.pyplot as plt
import pickle

fig, axes = plt.subplots(ncols=3, sharey="row")
axes[0].plot([0,1],[0,1])
axes[1].plot([0,1],[1,2])
axes[2].plot([0,1],[2,3])

pickle.dump(fig, file('fig1.pkl', 'wb'))  

plt.close("all")

fig2 = pickle.load(file('fig1.pkl','rb'))
ax_master = fig2.axes[0]
for ax in fig2.axes:
    if ax is not ax_master:
        ax_master.get_shared_y_axes().join(ax_master, ax)

plt.show()

它的缺点是您需要知道共享哪些轴。如上所示,在共享所有轴的情况下,这当然很容易。