matplotlib__examples/examples/exm16.py - master - ilya.petrash/SimplePyScripts

matplotlib__examples/examples/exm16.py
57 строк3 KB
gil9red
Refactoring. Using black code style
30 май 2023, 15:01
30 май 2023, 15:015886cad
__author__ = "ipetrash"


# Использование библиотеки Matplotlib. Как отображать формулы в нотации TeX
# Одной из интересных особенностей Matplotlib является то, что эта библиотека умеет
# рисовать формулы по их TeX-нотации. Чтобы указать Matplotlib, что строка содержит
# формулу, достаточно эту формулу заключить в знаки $.
# Допустим, мы хотим, чтобы в заголовке легенды у нас располагалась формула,
# описывающая функцию sinc (f(x) = sin (x) / x). Для этого достаточно в качестве
# параметра title функции legend() передать параметр r"$f(x) = \frac{sin(x)}{x}$".
# Здесь удобно использовать "сырые" строки с префиксом "r", чтобы не приходилось
# каждый раз удваивать обратный слеш, который в TeX обозначает начало команды.


import math

# Импортируем один из пакетов Matplotlib
import pylab

# Импортируем пакет со вспомогательными функциями
from matplotlib__examples import mlab


# Будем рисовать график этой функции
def func(x):
    """
    sinc (x)
    """
    if x == 0:
        return 1.0
    return math.sin(x) / x


if __name__ == "__main__":
    # Интервал изменения переменной по оси X
    xmin = -20.0
    xmax = 20.0

    # Шаг между точками
    dx = 0.01

    # Создадим список координат по оиси X на отрезке [-xmin; xmax], включая концы
    xlist = mlab.frange(xmin, xmax, dx)

    # Вычислим значение функции в заданных точках
    ylist1 = [func(x) for x in xlist]
    ylist2 = [func(x * 0.2) for x in xlist]

    # !!! Графики имеют метки с формулами в формате TeX
    pylab.plot(xlist, ylist1, "b-", label="$f(x)$")
    pylab.plot(xlist, ylist2, "g--", label="$f(x \cdot 0.2)$")

    # !!! Добавим легенду с заголовком в виде формулы
    pylab.legend(title=r"$f(x) = \frac{sin(x)}{x}$")

    # Покажем окно с нарисованным графиком
    pylab.show()