python - matplotlib:绘图完成后释放内存

Question

我想知道在绘图完成后释放内存的正确方法，因为在循环中绘制多个图像时出现 RuntimeError: Could not allocate memory for image 错误。

目前我在另一个代码中有以下命令可以做到这一点:

import matplotlib.pyplot as plt

# The code
.....

# Make plot
fig = plt.figure()
# Plotting stuff.
plt.imshow(...)
plt.plot(...)
plt.scatter(...)

# Save plot to file.
plt.savefig(...)

# Release memory.
plt.clf()
plt.close()

this answer 中的评论声明正确的语法实际上是 plt.close(fig) 但投票最高的答案是 given here说 plt.clf() 就足够了，没有提到 .close。

问题是:绘图保存到文件后释放内存的正确命令是什么？我需要同时使用 .clf 和 .close 还是其中一个就足够了？

Answer 1

我想为您推荐另一种方法。请注意，imshow 会为您返回句柄。获取对此的引用，并在该对象上使用 set_data 方法进行后续迭代。

>>> h = plt.imshow(np.zeros([480, 640]))
>>> h
<matplotlib.image.AxesImage at 0x47a03d0>
>>> for img in my_imgs:
...     h.set_data(img)  #etc