存储并重新加载 matplotlib.pyplot 对象

Question

我在一个伪操作环境中工作，我们在收到数据后制作新图像。 ，当新数据进来时，我们需要重新打开图像并更新该图像，以便创建合成、添加叠加等。除了添加到图像之外，这还需要修改标题、图例等。

有时 matplotlib 中内置的东西可以让我存储和重新加载 matplotlib.pyplot 对象以供以后使用？它需要维护对所有关联对象的访问，包括图形、线条、图例等。也许 pickle 就是我正在寻找的，但我对此表示怀疑。

Answer 1

从 1.2 开始，matplotlib 附带了实验性酸洗支持。如果您遇到任何问题，请在 mpl 邮件列表上告知我们，或者在 github.com/matplotlib/matplotlib 上提出问题

HTH

编辑：添加了一个简单的示例，

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

ax = plt.subplot(111)
x = np.linspace(0, 10)
y = np.exp(x)
ax.plot(x, y)
pickle.dump(ax, open('myplot.pickle', 'w'))

然后在单独的会话中:

import matplotlib.pyplot as plt
import pickle

ax = pickle.load(open('myplot.pickle'))
plt.show()

Answer 2

对 Pelson 针对在 Jupyterhub 上工作的人员的答案进行了一个小修改，

在加载 pickle 之前使用 %matplotlib notebook 。在 jupyterhub 或 jupyter 笔记本中使用 %matplotlib inline 对我来说不起作用。并给出以结尾的回溯
AttributeError：“模块”对象没有属性“new_figure_manager_given_figure”。

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

%matplotlib notebook

ax = plt.subplot(111)
x = np.linspace(0, 10)
y = np.exp(x)
plt.plot(x, y)
with open('myplot.pkl','wb') as fid:
    pickle.dump(ax, fid)

然后在一个单独的会话中：

import matplotlib.pyplot as plt
import pickle

%matplotlib notebook

with open('myplot.pkl','rb') as fid:
    ax = pickle.load(fid)
plt.show()

Answer 3

我使用 matplotlib 为多篇论文生成了图表。我不会考虑保存图形（如在 MATLAB 中），而是编写一个脚本来绘制数据，然后格式化并保存图形。如果我想保留数据的本地副本（特别是如果我想能够再次使用它），我发现 numpy.savez() 和 numpy.load() 非常有用。

起初，我怀念在 MATLAB 中保存图形的收缩包装感觉，但一段时间后，我开始喜欢这种方法，因为它包含可用于进一步分析的格式的数据。

Answer 4

在 jupyter 笔记本上测试。

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

fig, axes = plt.subplots(figsize=(20, 5),nrows=1, ncols=2) 
x,y = np.arange(10), np.random.random(10)
axes[0].plot(x,y)
axes[1].plot(x,y)
plt.show()

# myfname = input("Save figure? [enter filename]: ") 
myfname = 'test'
if (myfname!=''):
    fig.savefig(f'./{myfname}.png')
    print(f'file saved to ./{myfname}.png')
    with open(f'./{myfname}.png.pkl','wb') as fid:
        pickle.dump(fig, fid)
        print(f'pickled to ./{myfname}.png.pkl') 

###################################
####### in a separate session
myfname = 'test'
with open(f'./{myfname}.png.pkl', 'rb') as fh:
    fig_loaded = pickle.load(fh)

print(fig_loaded.axes[0].lines[0].get_data()) # get data
fig_loaded # show fig

Answer 5

你尝试过pickle模块吗？它序列化一个对象，将其转储到文件中，并可以稍后从文件中重新加载它。