如何使 pyplot.close() 在 python 中正常工作？

Question

我想创作很多，至少10000个情节， 所以我用for循环来做，这是我的代码。

for i in range(0, 10000):
    fig, ax = plt.subplots(1, 1, num=0)
    x = np.linspace(0, 100, 100)
    y = np.sin(x)
    
    ax.plot(x, y)
    plt.tight_layout()
    fig.savefig("test.png")
    plt.close(fig=0)
    
def plot_test():
    for i in range(0, 10000):
        fig, ax = plt.subplots(1, 1, num=0)
        x = np.linspace(0, 100, 100)
        y = np.sin(x)
        
        ax.plot(x, y)
        plt.tight_layout()
        fig.savefig("test.png")
        plt.close(fig=0)

无论在函数内部还是外部使用 plt.close()， 内存还在增加， 我检查了 plt.close("all") 和 plt.close(fig) 也没有工作。 希望有人能告诉我如何正确使用 plt.close()。

Answer 1

所以我相信plt.close() 工作正常。您的问题是每个情节都具有相同的名称。这将覆盖目录中的任何现有文件。本质上，它创建test.png 并在每次迭代时立即覆盖它。此解决方案将i 添加到绘图名称的末尾，并保存到我创建的名为test 的目录中。有了这么多的地块，完成迭代需要一点时间

import numpy as np
import matplotlib
# The default backend is "tkinter" which is used for displaying the graphs. 
# "agg" is more lightweight and avoids a tkinter error when making lots of plots
matplotlib.use('agg')
import matplotlib.pyplot as plt
import gc



for i in range(0, 10000):
    # This just shows the progress of the loop
    print(i)
   
    fig, ax = plt.subplots(1, 1, num=0)
    x = np.linspace(0, 100, 100)
    y = np.sin(x)

    ax.plot(x, y)
    plt.tight_layout()
    fig.savefig("test/test{}.png".format(str(i)))
    plt.close(fig=fig)
    gc.collect()