Создание трехмерных контурных визуализаций

Beginner

This tutorial is from open-source community. Access the source code

Введение

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

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

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

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

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

Импорт библиотек

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

import matplotlib.pyplot as plt
from matplotlib import cm
from mpl_toolkits.mplot3d import axes3d

Создание фигуры и подграфика

Далее создадим фигуру и подграфик для размещения нашего трехмерного контурного графика.

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

Получение тестовых данных

Мы будем использовать функцию axes3d.get_test_data() для получения некоторых тестовых данных для построения графика.

X, Y, Z = axes3d.get_test_data(0.05)

Создание контурного графика

Теперь мы можем создать трехмерный контурный график тестовых данных с использованием функции ax.contour().

ax.contour(X, Y, Z, cmap=cm.coolwarm)

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

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

ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
ax.view_init(elev=30, azim=120)

Показать график

Наконец, мы используем функцию plt.show(), чтобы показать трехмерный контурный график.

plt.show()

Резюме

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