Использование библиотеки Matplotlib. Как отображать формулы в нотации TeX
Одной из интересных особенностей Matplotlib является то, что эта библиотека умеет рисовать формулы по их TeX-нотации. Чтобы указать Matplotlib, что строка содержит формулу, достаточно эту формулу заключить в знаки $.
Допустим, мы хотим, чтобы в заголовке легенды у нас располагалась формула, описывающая функцию sinc (f(x) = sin (𝜋x) / (𝜋x)). Для этого достаточно в качестве параметра title функции legend() передать параметр r"$f(x) = \frac{sin(\pi x)}{(\pi x)}$".
Здесь удобно использовать "сырые" строки с префиксом "r", чтобы не приходилось каждый раз удваивать обратный слеш, который в TeX обозначает начало команды.
Точно также можно использовать формулы и в других выводимых строках. Это показано в следующем примере:
# Импортируем один из пакетов Matplotlib
import matplotlib.pyplot as plt
import numpy as np
if __name__ == '__main__':
# Интервал изменения переменной по оси X
xmin = -20.0
xmax = 20.0
# Шаг между точками
dx = 0.01
# Создадим список координат по оиси X на отрезке [xmin; xmax], включая концы
x = np.arange(xmin, xmax + dx, dx)
# Вычислим значение функции в заданных точках
y1 = np.sinc(x)
y2 = np.sinc(x * 0.2)
# !!! Графики имеют метки с формулами в формате TeX
plt.plot (x, y1, "b-", label=r"$f(x)$")
plt.plot (x, y2, "g--", label=r"$f(x \cdot 0.2)$")
# !!! Добавим легенду с заголовком в виде формулы
plt.legend (title=r"$f(x) = \frac{sin(\pi x)}{(\pi x)}$")
# Покажем окно с нарисованным графиком
plt.show()
Вы можете подписаться на новости сайта через RSS, Группу Вконтакте или Канал в Telegram.
