将多个图存储在列表中以便以后访问它们

Question

我在我的 python 文件中创建了几个带有 matplotlib 的图。 我想稍后访问它们，然后在 VTK object 上访问它们。 我试图创建一个列表，只是 append 它们作为 plot 对象。

for i in range(len(x)):
    ax = plt.subplot(111, projection='polar')
    plt.legend()
    plot_l.append(plt.show())

但这并不能很好地工作。 我还尝试使用plt.savefig('somewhere/folder')并将它们从文件中重新加载。

但这并不能满足我的需求，因为我有大量的地块，我想把这些地块放在文件中。

如何将多个图存储为png以便以后访问它们，例如plot_l[10] ？ 还是有更好的选择？

干杯，

Answer 1

您可能想尝试保存您的ax object，而不是。

import pickle

# [...your plotting code here...]

pickle.dump(ax, file('filename', 'w'))

这就是您下次需要时加载它的方式：

ax = pickle.load(file('filename'))

在您的特定情况下，您可能更喜欢将其放在循环中，并且filename取决于循环索引等。

最后，您可能想尝试使用列表 object 进行相同操作，以将所有内容保存在一个文件中，如您所指示的。