Добавление вспомогательных осей к графикам в Matplotlib

Улучшение графиков в Matplotlib с помощью добавления вспомогательных осей

Как добавить вставные оси к фигуре Matplotlib для улучшения визуализации данных

Изображение, созданное DALL-E 3 автором.

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

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

В этом кратком руководстве мы рассмотрим, как создать простую карту местоположения скважин, подобную показанной ниже, с использованием координат сетки. Затем мы добавим вставные оси, чтобы выделить и увеличить меньшую область на главной карте.

Финальная фигура matplotlib, содержащая вставные оси, подсвечивает меньшую выборку скважин. Изображение автора.

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

Первый шаг в нашем руководстве – импортировать используемые библиотеки.

Первые два оператора импорта должны быть знакомы: импорт pandas и matplotib. Третий импорт из mpl.toolkits.axes_grid1.inset_locator позволяет нам визуализировать вставную фигуру в пределах нашей основной фигуры.

import pandas as pdimport matplotlib.pyplot as pltfrom mpl_toolkits.axes_grid1.inset_locator import inset_axes, mark_inset

После импорта библиотек мы можем импортировать данные. Используемые здесь данные взяты из соревнования по машинному обучению Force 2020 Xeek и содержат около 100 скважин с измерениями.

Поскольку файл разделен точкой с запятой ( ; ), нам нужно указать дополнительный параметр в функции pd.read_csv(), чтобы она знала, что это разделитель между значениями.

df = pd.read_csv('../data/Xeek Force 2020/train.csv', sep=';')

Создание карты местонахождения скважины с использованием координат X и Y в Matplotlib