Question

我正在编写代码以创建图并使用matplotlib中的plt.savefig方法保存它们。

我想知道此函数如何处理文件处理问题（准确地说是关闭文件句柄），并确保在创建大量此类图时是否不会引起内存管理问题。

我尝试查找MatPlotLib的代码库以及与此有关的一些StackOverflow问题，但尚未发现任何东西。

我的猜测是，由于我没有创建文件处理程序，因此不存在任何要关闭的引用。但是，我想确定这个假设，并想了解后端到底发生了什么。

    data_array = np.random.normal(size=10)
    plt.boxplot(data_array)
    plt.savefig(os.path.join(path_to_file,"box_plot.png"))
    plt.close()

Answer 1

这取决于您实际使用哪个后端绘制图像。从您发布的代码片段中，matplotlib会猜测应该使用PNG后端。 依次创建matplotlib.backends.backend_agg.FigureCanvasAgg的实例。

此类实际保存如下：

def print_raw(self, filename_or_obj, *args, **kwargs):
    FigureCanvasAgg.draw(self)
    renderer = self.get_renderer()
    with cbook._setattr_cm(renderer, dpi=self.figure.dpi), \
            cbook.open_file_cm(filename_or_obj, "wb") as fh:
        fh.write(renderer._renderer.buffer_rgba())

因此，没有悬空的打开文件。该文件保证可以正确关闭。 看看the source code。