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

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


# Использование библиотеки Matplotlib. Как изменять размеры ячеек таблицы при
# расположении графиков в окне.
#
# Продолжим разбираться с новыми возможностями Matplotlib, появившимися в версии
# 1.0.0. В этом примере посмотрим как с помощью этого класса GridSpec сделать так,
# чтобы размеры ячеек для графиков были бы разных размеров. Причем, в данном случае
# речь будет идти не о том, что графики занимают несколько ячеек, регулирование
# размеров будет осуществляться именно за счет размеров ячеек.


import math

import pylab

import matplotlib__examples


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

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

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

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

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

    pylab.plot(xlist, ylist)


if __name__ == "__main__":
    # Создаем таблицу (сетку) с тремя строками и двумя столбцами
    rows = 3
    cols = 2

    # При создании сетки появились два новых параметра:
    # width_ratios - список соотношений ширин ячеек.
    # В данном случае ширина второго столбца будет в 1.5 раза больше первого
    # height_ratios - список соотношений высот ячеек
    # В данном случае высота второй строки будет в 2 раза больше первого,
    # а высота третьей строки будет в 3 раза больше первой (в 1.5 раза больше второй)
    grid = matplotlib__examples.gridspec.GridSpec(
        rows, cols, width_ratios=[1, 1.5], height_ratios=[1, 2, 3]
    )

    for n in range(rows * cols):
        pylab.subplot(grid[n])
        plotGraph()

    pylab.show()