Создание поддиаграмм в Matplotlib для Python

Введение

Matplotlib - это библиотека на Python, используемая для создания визуализаций, таких как линейные диаграммы, точечные диаграммы, столбчатые диаграммы и поддиаграммы и другие. В этом лабе вы научитесь создавать фигуру с двумя поддиаграммами с использованием .pyplot.subplot.

Советы по ВМ

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

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

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

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

Для этого руководства мы будем использовать модуль pyplot из библиотеки Matplotlib и библиотеку numpy.

import matplotlib.pyplot as plt
import numpy as np

Определяем данные

Мы определим два набора данных, которые будем использовать для создания наших поддиаграмм.

def f(t):
    return np.exp(-t) * np.cos(2*np.pi*t)

t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)

Создаем поддиаграммы

Мы создадим фигуру с двумя поддиаграммами с использованием .pyplot.subplot.

plt.figure()

plt.subplot(211)
plt.plot(t1, f(t1), color='tab:blue', marker='o')
plt.plot(t2, f(t2), color='black')

plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), color='tab:orange', linestyle='--')

plt.show()

Функция subplot() принимает три аргумента: количество строк, количество столбцов и индекс текущей диаграммы. Индекс начинается с 1 в верхнем левом углу и увеличивается по строкам. В этом примере мы создаем фигуру с двумя поддиаграммами: одну сверху и одну снизу.

В первой поддиаграмме мы строим график зависимости t1 от f(t1) и t2 от f(t2). Мы задаем цвет первой диаграммы синим и добавляем круглые маркеры к каждой точке данных. Мы задаем цвет второй диаграммы черным.

Во второй поддиаграмме мы строим график зависимости t2 от косинусной функции 2*np.pi*t2. Мы задаем цвет диаграммы оранжевым и стиль линии пунктирным.

Ссылки

В этом примере показано использование следующих функций, методов, классов и модулей:

• matplotlib.pyplot.figure
• matplotlib.pyplot.subplot

Резюме

В этом практическом занятии мы узнали, как создать фигуру с двумя поддиаграммами с использованием .pyplot.subplot. Мы определили два набора данных и построили их в двух поддиаграммах, используя разные цвета и стили линий. Мы также предоставили ссылки на функции и модули, использованные в этом примере.