python - 保存许多图后 Matplotlib 崩溃

Question

我正在为以后的动画绘制并保存数千个文件，如下所示:

import matplotlib.pyplot as plt
for result in results:
    plt.figure()
    plt.plot(result)                     # this changes
    plt.xlabel('xlabel')                 # this doesn't change
    plt.ylabel('ylabel')                 # this doesn't change
    plt.title('title')                   # this changes
    plt.ylim([0,1])                      # this doesn't change
    plt.grid(True)                       # this doesn't change
    plt.savefig(location, bbox_inches=0) # this changes

当我运行此程序并获得大量结果时，它会在保存几千个绘图后崩溃。我想我想做的是像这个答案一样重用我的轴:https://stackoverflow.com/a/11688881/354979但我不明白如何。我该如何优化它？

Answer 1

我每次都会创建一个图形并清除图形(使用 .clf)。

import matplotlib.pyplot as plt

fig = plt.figure()

for result in results:
    fig.clf()   # Clears the current figure
    ...

您的内存不足，因为每次调用 plt.figure 都会创建一个新的图形对象。根据@tcaswell 的评论，我认为这会比 .close 更快。差异解释如下:

When to use cla(), clf() or close() for clearing a plot in matplotlib?