Учебник по созданию подграфиков в Matplotlib

Введение

В этом руководстве вы узнаете, как использовать функцию subplot2grid из библиотеки Matplotlib для Python для генерации подграфиков.

Советы по работе с ВМ

После завершения запуска ВМ нажмите в левом верхнем углу, чтобы переключиться на вкладку Ноутбук и получить доступ к Jupyter Notebook для практики.

Иногда вам может потребоваться подождать несколько секунд, пока Jupyter Notebook не загрузится полностью. Валидация операций не может быть автоматизирована из-за ограничений Jupyter Notebook.

Если вы сталкиваетесь с проблемами во время обучения, не стесняйтесь обращаться к Labby. Оставьте отзыв после занятия, и мы оперативно решим проблему для вас.

Импортируем необходимые библиотеки

Прежде чем мы начнем, нам нужно импортировать библиотеку Matplotlib с помощью следующего кода:

import matplotlib.pyplot as plt

Создаем объект Figure

Для создания объекта Figure используйте следующий код:

fig = plt.figure()

Определяем подграфики с использованием subplot2grid

Для определения подграфиков с использованием subplot2grid сначала необходимо указать размер сетки с помощью кортежа, содержащего желаемое количество строк и столбцов. Также необходимо указать расположение подграфика внутри сетки с помощью другого кортежа.

Например, чтобы создать 3x3 сетку с подграфиком, который занимает всю первую строку и все три столбца, используем следующий код:

ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=3)

Для создания подграфика, который занимает вторую строку и первые два столбца, используем:

ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2)

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

ax3 = plt.subplot2grid((3, 3), (1, 2), rowspan=2)

Для создания подграфика в последней строке и первом столбце используем:

ax4 = plt.subplot2grid((3, 3), (2, 0))

Для создания подграфика в последней строке и втором столбце используем:

ax5 = plt.subplot2grid((3, 3), (2, 1))

Аннотируем оси

Для аннотации осей мы можем пройтись по осям фигуры и добавить текст с использованием функции text и функции tick_params для удаления меток делений.

def annotate_axes(fig):
    for i, ax in enumerate(fig.axes):
        ax.text(0.5, 0.5, "ax%d" % (i+1), va="center", ha="center")
        ax.tick_params(labelbottom=False, labelleft=False)

Отображаем график

Для отображения графика используйте следующий код:

plt.show()

Резюме

В этом уроке вы узнали, как использовать функцию subplot2grid в Matplotlib для генерации подграфиков. Также вы узнали, как создать объект фигуры, определить подграфики внутри сетки, аннотировать оси и отобразить график.