Использование библиотеки Matplotlib. Как отображать формулы в нотации TeX

Одной из интересных особенностей Matplotlib является то, что эта библиотека умеет рисовать формулы по их TeX-нотации. Чтобы указать Matplotlib, что строка содержит формулу, достаточно эту формулу заключить в знаки $.

Допустим, мы хотим, чтобы в заголовке легенды у нас располагалась формула, описывающая функцию sinc (f(x) = sin (𝜋x) / (𝜋x)). Для этого достаточно в качестве параметра title функции legend() передать параметр r"$f(x) = \frac{sin(\pi x)}{(\pi x)}$".

Здесь удобно использовать "сырые" строки с префиксом "r", чтобы не приходилось каждый раз удваивать обратный слеш, который в TeX обозначает начало команды.

Точно также можно использовать формулы и в других выводимых строках. Это показано в следующем примере:

# -*- coding: utf-8 -*-

# Импортируем один из пакетов 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()

Другие статьи про Matplotlib

Вы можете подписаться на новости сайта через RSS, Группу Вконтакте или Канал в Telegram.
5 stars

Рейтинг 4.6/5. Всего 8 голос(а, ов)




Подписаться на комментарии
Автор:
Тема:
 Ваш комментарий
 
 
Введите код 378