Добавить цветовую шкалу на вставленные оси | Руководство по Matplotlib для Python

Введение

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

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

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

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

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

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

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

import matplotlib.pyplot as plt
from matplotlib import cbook
from mpl_toolkits.axes_grid1.inset_locator import inset_axes, zoomed_inset_axes

fig, ax = plt.subplots(figsize=[5, 4])

Z = cbook.get_sample_data("axes_grid/bivariate_normal.npy")
extent = (-3, 4, -4, 3)

Настройка основного графика

Настройте основной график, настроив соотношение сторон и пределы по осям x и y.

ax.set(aspect=1, xlim=(-15, 15), ylim=(-20, 5))

Создание вставленной оси

Создайте вставленную ось с помощью функции zoomed_inset_axes. Задайте уровень приближения и расположение вставленной оси внутри основного графика.

axins = zoomed_inset_axes(ax, zoom=2, loc='upper left')
axins.set(xticks=[], yticks=[])

Добавление изображения на вставленную ось

Добавьте изображение на вставленную ось с помощью функции imshow. Задайте размерность и начало изображения.

im = axins.imshow(Z, extent=extent, origin="lower")

Добавление цветовой шкалы

Добавьте цветовую шкалу на вставленную ось с помощью функции inset_axes. Задайте ширину, высоту, расположение и ограничивающий прямоугольник цветовой шкалы.

cax = inset_axes(axins,
                 width="5%",  ## width = 10% of parent_bbox width
                 height="100%",  ## height : 50%
                 loc='lower left',
                 bbox_to_anchor=(1.05, 0., 1, 1),
                 bbox_transform=axins.transAxes,
                 borderpad=0,
                 )
fig.colorbar(im, cax=cax)

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

Отобразите график с использованием функции show.

plt.show()

Резюме

Поздравляем! Вы успешно научились добавлять цветовую шкалу на вставленные оси с использованием Matplotlib в Python. Это полезный метод для визуализации данных более детально и информативно. Не забудьте настраивать параметры в соответствии с вашими конкретными потребностями и предпочтениями.

Добавление цветовой шкалы на вставленные оси