Сохраните график matplotlib в файл с помощью командной строки без использования plt.save ()

Question

У меня есть скрипт, который создает график, а затем показывает его:

(следующий скрипт - пример из документов )

# coding: utf-8
# file: test_plot.py

import yaml

import matplotlib.pyplot as plt
import numpy as np


def plot():
    t = np.arange(0.0, 2.0, 0.01)
    s = 1 + np.sin(2*np.pi*t)
    plt.plot(t, s)

    plt.xlabel('time (s)')
    plt.ylabel('voltage (mV)')
    plt.title('About as simple as it gets, folks')
    plt.grid(True)
    plt.show()

if __name__ == '__main__':
    plot()

Если этоСценарий выполняется из командной строки (python test_plot.py), он корректно отображает сюжет.

Вопрос: можно сохранить сюжет в файл без изменения кода?

Answer 1

Без изменения кода вы, конечно, не можете изменить его результат.Поэтому я буду интерпретировать это как «без изменения функции plot».Т.е. вы можете изменить все, что ниже if __name__ == '__main__':.

В этом случае вы можете включить интерактивный режим при вызове функции и впоследствии отключить ее.

if __name__ == '__main__':
    plt.ion()
    plot()
    plt.ioff()
    plt.savefig("trala.png")
    plt.show()