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