Как использовать LaTeX в Python Matplotlib: Полное руководство

Matplotlib является де-факто стандартом для визуализации данных в Python, предоставляя гибкие инструменты для создания разнообразных графиков. LaTeX, в свою очередь, — это система верстки высокого уровня, широко используемая в академических кругах для набора текста, особенно математических формул, благодаря непревзойденному качеству типографики.

Интеграция LaTeX в Matplotlib позволяет использовать мощь LaTeX для рендеринга текстовых элементов графиков, таких как подписи осей, заголовки, легенды и аннотации.

Зачем использовать LaTeX в Matplotlib?

Основная причина — получение высококачественных, профессионально выглядящих графиков, особенно при наличии сложных математических выражений. LaTeX обеспечивает:

Превосходное отображение формул: Ни один другой инструмент не сравнится с LaTeX в рендеринге математики.

Консистентность шрифтов: Возможность использовать те же шрифты, что и в основном документе (например, статье или отчете), созданном с помощью LaTeX.

Доступ к символам и макросам: Использование всего богатства символов и макросов LaTeX.

Установка LaTeX и необходимых пакетов (если требуется)

Для использования LaTeX в Matplotlib необходимо иметь рабочую установку LaTeX на вашей системе. Популярные дистрибутивы:

TeX Live: Кроссплатформенный, рекомендован для Linux и macOS.

MiKTeX: В основном для Windows, но доступен и для других ОС.

MacTeX: Специализированная версия TeX Live для macOS.

Кроме самого дистрибутива LaTeX, Matplotlib обычно требует dvipng (для рендеринга в PNG) и Ghostscript. Они могут быть установлены через менеджер пакетов вашего дистрибутива LaTeX или системный менеджер пакетов.

Проверить наличие LaTeX можно, выполнив latex -v в терминале.

Основы работы с Matplotlib

Предполагается, что читатель знаком с основами Matplotlib: создание фигуры и осей (plt.figure, plt.subplots), построение базовых графиков (ax.plot, ax.scatter), настройка основных элементов (ax.set_xlabel, ax.set_title) и отображение графика (plt.show).

Базовое использование LaTeX в Matplotlib

Чтобы активировать рендеринг текста через LaTeX, необходимо установить соответствующий параметр в конфигурации Matplotlib. Рекомендуется делать это в начале скрипта.

import matplotlib.pyplot as plt
import numpy as np
from typing import Dict

# Активация LaTeX для рендеринга текста
plt.rcParams.update({
    "text.usetex": True,
    "font.family": "serif", # Рекомендуется использовать serif для лучшей совместимости с LaTeX
    "font.serif": ["Computer Modern Roman"], # Классический шрифт LaTeX
})

# Пример данных (например, метрики рекламной кампании)
x: np.ndarray = np.linspace(0, 10, 100)
y_ctr: np.ndarray = 0.5 * np.sin(x) + 1 # Пример CTR
y_cost: np.ndarray = 0.2 * x + 0.5 # Пример затрат

fig, ax = plt.subplots()
ax.plot(x, y_ctr)

Важно использовать "сырые" строки (raw strings) r"..." для передачи LaTeX-команд Matplotlib, чтобы избежать проблем с экранированием обратных слешей.

Использование LaTeX для текста подписей осей (xlabel, ylabel)

Просто передайте LaTeX-форматированную строку в ax.set_xlabel() или ax.set_ylabel().

# ... (продолжение предыдущего примера)

ax.set_xlabel(r'Время $t$, дни')
ax.set_ylabel(r'CTR, %')

Использование LaTeX для заголовка графика (title)

Аналогично подписям осей, используйте ax.set_title().

# ... (продолжение предыдущего примера)

ax.set_title(r'Динамика CTR кампании $\alpha \rightarrow \beta$') # Используем греческие буквы

Добавление LaTeX формул в текст графика (text)

Функция ax.text() позволяет размещать текст в произвольной точке графика. Это идеально подходит для добавления формул или аннотаций.

# ... (продолжение предыдущего примера)

# Добавим аннотацию с формулой средней стоимости
mean_cost: float = np.mean(y_cost)
ax.text(2, 1.4, fr'$\bar{{C}} = {mean_cost:.2f}$ у.е.', fontsize=12) # Используем f-string вместе с raw string

# plt.show() # Отобразить график

Продвинутое использование LaTeX в Matplotlib

Использование LaTeX в легенде графика (legend)

LaTeX можно использовать в метках данных, передаваемых через параметр label в функциях построения графиков (plot, scatter и т.д.).

import matplotlib.pyplot as plt
import numpy as np

# Настройки LaTeX
plt.rcParams.update({
    "text.usetex": True,
    "font.family": "serif",
    "font.serif": ["Computer Modern Roman"],
})

x: np.ndarray = np.linspace(0, 2 * np.pi, 100)
y1: np.ndarray = np.sin(x)
y2: np.ndarray = np.cos(x)

fig, ax = plt.subplots()

# Используем LaTeX в метках
ax.plot(x, y1, label=r'$y = \sin(x)$')
ax.plot(x, y2, label=r'$y = \cos(x)$')

# Отображаем легенду
ax.legend(fontsize=11)

ax.set_xlabel(r'$x$')
ax.set_ylabel(r'$f(x)$')
ax.set_title(r'Тригонометрические функции')

# plt.show()

Настройка шрифтов и размеров LaTeX

Глобальные настройки шрифта (семейство, размер) задаются через rcParams. Для более тонкой настройки (например, подключения специфичных LaTeX пакетов) используется параметр text.latex.preamble.

plt.rcParams.update({
    "text.usetex": True,
    "font.family": "sans-serif",
    "font.sans-serif": ["Helvetica"], # Пример использования sans-serif шрифта
    "font.size": 12,
    "text.latex.preamble": r"""
        \usepackage{amsmath}      % для расширенных математических окружений
        \usepackage{amsfonts}     % для математических шрифтов (mathbb)
        \usepackage[T2A]{fontenc} % поддержка кириллицы (если нужна)
        \usepackage[utf8]{inputenc}% кодировка utf8
    """
})

# Пример использования AMSmath
fig, ax = plt.subplots()
ax.text(0.5, 0.5, r'$\mathbb{E}[X] = \int_{-\infty}^{\infty} x f(x) \, dx$', 
        horizontalalignment='center', verticalalignment='center', fontsize=16)
ax.set_xticks([])
ax.set_yticks([])
# plt.show()

Использование LaTeX для математических символов и формул

LaTeX предоставляет огромный набор команд для математических символов: греческие буквы (\alpha, \beta), операторы (\sum, \int, \prod), дроби (\frac{a}{b}), корни (\sqrt{x}), степени и индексы (^, _), и многое другое.

Вставка сложных математических выражений

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

# ... (Настройки rcParams с amsmath)

fig, ax = plt.subplots()

# Пример системы уравнений
eq_system: str = r"""\begin{cases}
   a_{11}x_1 + a_{12}x_2 = b_1 \\
   a_{21}x_1 + a_{22}x_2 = b_2
\end{cases}"""

ax.text(0.5, 0.5, eq_system, 
        horizontalalignment='center', verticalalignment='center', fontsize=14)
ax.set_xticks([])
ax.set_yticks([])
ax.set_title('Система линейных уравнений')

# plt.show()

Решение проблем и отладка

Распространенные ошибки и способы их устранения

RuntimeError: Failed to process string... или RuntimeError: latex was not able to process the following string...:

Убедитесь, что LaTeX установлен и доступен в системном PATH.

Проверьте наличие dvipng и Ghostscript.

Проверьте синтаксис LaTeX в вашей строке. Попробуйте скомпилировать минимальный LaTeX-документ с этой строкой вручную.

Проверьте логи Matplotlib (могут содержать вывод LaTeX).

Квадраты вместо символов / неверное отображение:

Проблема с кодировкой или шрифтами. См. следующий пункт.

Проблемы с кодировкой и шрифтами

Используйте UTF-8 в ваших Python скриптах (# -*- coding: utf-8 -*- в начале файла для Python 2, в Python 3 это стандарт).

Настройте преамбулу LaTeX (text.latex.preamble) для поддержки нужной кодировки (inputenc) и шрифтов (fontenc). Для кириллицы часто требуются \usepackage[T2A]{fontenc} и \usepackage[utf8]{inputenc}.

Убедитесь, что выбранный шрифт (font.family, font.serif/font.sans-serif) поддерживает необходимые символы и установлен в системе и/или доступен LaTeX.

Медленная отрисовка графиков с LaTeX

Рендеринг через LaTeX значительно медленнее встроенного механизма Matplotlib (mathtext).

Кеширование: Matplotlib кеширует результаты компиляции LaTeX. При повторном использовании тех же строк рендеринг будет быстрее. Очистить кеш можно удалив ~/.matplotlib/tex.cache.

Сложность: Избегайте чрезмерно сложных LaTeX-выражений в элементах, которые часто обновляются (например, в интерактивных графиках).

Альтернативы: Для простых формул используйте mathtext (установив text.usetex: False). Он быстрее, но имеет ограниченные возможности по сравнению с полным LaTeX.

Примеры и лучшие практики

Примеры графиков с использованием LaTeX (разные типы графиков)

LaTeX можно эффективно использовать в любых типах графиков:

Линейные графики: Аннотации с производными (\frac{dy}{dx}), интегралами (\int y \, dx), обозначениями точек (P_i).

Гистограммы/Распределения: Подписи с параметрами распределений (\mathcal{N}(\mu, \sigma^2)), формулами плотности вероятности.

Диаграммы рассеяния: Уравнения регрессионных моделей (y = \beta_0 + \beta_1 x + \epsilon), коэффициенты корреляции (\rho_{xy}).

Советы по оформлению графиков для публикации

Консистентность: Используйте те же шрифты и размеры, что и в основном тексте вашей публикации.

Читаемость: Не перегружайте график формулами. Используйте аннотации умеренно.

Векторный формат: Сохраняйте графики в векторных форматах (PDF, SVG, EPS) для наилучшего качества при масштабировании. LaTeX-текст будет сохранен как векторный.

# plt.savefig('my_plot.pdf', format='pdf', bbox_inches='tight')

Проверка: Всегда внимательно проверяйте итоговый вид графика перед публикацией.

Альтернативные способы использования LaTeX в Matplotlib

mathtext: Встроенный в Matplotlib парсер математических выражений, похожий на подмножество TeX. Активируется по умолчанию (text.usetex: False). Быстрее, но менее функционален. Поддерживает основные команды и символы.

plt.rcParams.update({"text.usetex": False})
plt.title(r'Простой режим: $\alpha > \beta$') # Использует mathtext
# plt.show()

Внешние инструменты: Можно генерировать сложные формулы как изображения (например, с помощью онлайн-редакторов LaTeX или локально) и вставлять их на график с помощью ax.imshow.