Сохранение рисунков Matplotlib в формате Base64: удобное руководство с примерами кода

Привет! Если вы хотите сохранить данные Matplotlib в формате Base64, вы попали по адресу. В этой статье блога мы рассмотрим несколько методов выполнения этой задачи, дополненные простыми для понимания примерами кода. Итак, приступим!

Метод 1: использование объекта FigureCanvas

Один простой способ сохранить фигуру Matplotlib в формате Base64 — использовать объект FigureCanvas. Вот пошаговый пример:

import matplotlib.pyplot as plt
from io import BytesIO
import base64
# Create a Matplotlib figure
fig, ax = plt.subplots()
ax.plot([1, 2, 3, 4], [10, 5, 20, 15])
# Create a BytesIO buffer
buffer = BytesIO()
# Save the figure to the buffer
fig.savefig(buffer, format='png')
# Reset the buffer position to the beginning
buffer.seek(0)
# Read the buffer content as Base64
image_base64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
# Print the Base64 encoded figure
print(image_base64)

В этом примере мы сначала создаем фигуру Matplotlib и отображаем некоторые данные. Затем мы создаем буфер BytesIOи сохраняем в него фигуру как изображение PNG. Затем мы сбрасываем позицию буфера в начало и читаем его содержимое как Base64 с помощью функции base64.b64encode(). Наконец, мы печатаем фигуру в кодировке Base64.

Метод 2: использование метода getvalue()

Другой метод предполагает использование метода getvalue()объекта FigureCanvasдля прямого получения изображения PNG в виде данных BytesIO. Вот пример:

import matplotlib.pyplot as plt
from io import BytesIO
import base64
# Create a Matplotlib figure
fig, ax = plt.subplots()
ax.plot([1, 2, 3, 4], [10, 5, 20, 15])
# Get the PNG image as BytesIO data
buffer = BytesIO()
fig.canvas.print_png(buffer)
# Reset the buffer position to the beginning
buffer.seek(0)
# Read the buffer content as Base64
image_base64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
# Print the Base64 encoded figure
print(image_base64)

Здесь мы создаем фигуру Matplotlib и отображаем данные, как и раньше. Затем вместо использования savefig()мы напрямую вызываем fig.canvas.print_png(), чтобы получить изображение PNG в виде данных BytesIO. Остальная часть кода аналогична первому методу.

Метод 3: использование функции to_base64()(Matplotlib 3.4+)

Начиная с версии Matplotlib 3.4, вы можете использовать функцию to_base64()для сохранения фигуры непосредственно в формате Base64. Вот пример:

import matplotlib.pyplot as plt
# Create a Matplotlib figure
fig, ax = plt.subplots()
ax.plot([1, 2, 3, 4], [10, 5, 20, 15])
# Save the figure as Base64
image_base64 = fig.to_base64()
# Print the Base64 encoded figure
print(image_base64)

В этом методе мы создаем фигуру и отображаем данные. Затем мы просто вызываем fig.to_base64(), чтобы сохранить цифру в формате Base64. Это удобный и лаконичный способ выполнить задачу.

Заключение

В этой статье мы рассмотрели три метода сохранения рисунков Matplotlib в формате Base64. Мы использовали объект FigureCanvas, метод getvalue()и функцию to_base64(). Каждый метод предлагает немного отличающийся подход, но все они позволяют кодировать цифры в виде строк Base64 для различных целей.

Не забудьте импортировать необходимые библиотеки (matplotlib.pyplot, io.BytesIOи base64), прежде чем опробовать примеры кода.

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

Так что давай, попробуй! Приятного программирования и приятной визуализации!