Понимание того, как сохранять рисунки и графики matplotlib в файлы изображений

Введение

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

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

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

import matplotlib.pyplot as plt
plt.plot(
    [5, 4, 3], 
    [100, 200, 300] 
)
plt.title('Some Title')
plt.xlabel('Year')
plt.ylabel('Some measurements')
plt.show()

И созданная фигура показана ниже:

Сохранение графика на диске в виде файла изображения

Теперь, если вы хотите сохранить matplotlib фигур в виде файлов изображений программно, вам нужна только функция matplotlib.pyplot.savefig(). Просто укажите желаемое имя файла (и даже местоположение), и рисунок будет сохранен на вашем диске.

import matplotlib.pyplot as plt
plt.plot(
    [5, 4, 3], 
    [100, 200, 300] 
)
plt.title('Some Title')
plt.xlabel('Year')
plt.ylabel('Some measurements')
plt.savefig('my_plot.png')

Кроме того, вы все еще можете сохранить фигуру, щелкнув значок сохранения, который отображается в нижней части интерактивного окна при вызове plt.show().

plt.show()

Отключение интерактивного режима

В некоторых случаях рисунок все еще отображается в интерактивном режиме, даже когда plt.show() не вызывается после того, как мы сохранили график на свои диски. Это еще более раздражает, когда вы создаете несколько графиков, скажем, итеративно, так как вы получите множество открытых фигур, когда программа завершится.

Вы можете предотвратить это, просто вызвав метод matplotlib.pyplot.close(), который гарантирует закрытие окна рисунка.

import matplotlib.pyplot as plt
plt.plot(
 [5, 4, 3], 
 [100, 200, 300] 
)
plt.title('Some Title')
plt.xlabel('Year')
plt.ylabel('Some measurements')
plt.savefig('my_plot.png')
plt.close()

Последние мысли

В сегодняшней статье мы обсудили сохранение matplotlib рисунков и графиков в виде файлов изображений либо программно, либо с помощью инструментов, представленных в интерактивном окне, которое появляется при вызове plt.show().

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

Стать участником и читать все истории на Medium. Ваш членский взнос напрямую поддерживает меня и других писателей, которых вы читаете. Вы также получите полный доступ ко всем историям на Medium.



Вам также может понравиться