Динамически обновляемые графики в Python, как сохранить изображение в конце

Question

Фон

С помощью следующего я создал динамически обновляемый график

Динамически обновляемый график в matplotlib

Обновление сюжета в matplotlib Python

Как обновить сюжет в matplotlib?

Код сюжета

plt.ion()
fig_analysis = plt.figure()
for i in range(5):
    ax = fig_analysis.add_subplot(211)    
    l1, = ax.plot(np.linspace(0, 200, simulation_time), variable_1)
    ax.set_ylabel(r'$Variable\ 1$')
    ax = fig_analysis.add_subplot(212)
    l2, = ax.plot(np.linspace(0, 200, simulation_time), variable_2)
    ax.set_ylabel(r'$Variable\ 2$')
    ax.set_xlabel(r'$years$')
    fig_analysis.canvas.draw()
    plt.pause(0.5)

Поведение

Этот код создает график и обновляет его.Тем не менее, он закрывает окончательный график после завершения цикла

Вопрос

Как изменить код, чтобы в конце цикла программа не закрывала окно графикаи я могу сохранить изображение по своему усмотрению.

Решение вручную

Одним из способов достижения этого является ручное приостановление работы программы.Однако, поскольку время выполнения моей программы не фиксировано, реализовать эту стратегию сложно.

Answer 1

Добавить

plt.ioff()    # turn interactive mode off
plt.show()

в конце кода.