Как создать 3D-диаграммы рассеяния в Matplotlib

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

К концу этого урока вы научитесь:

Как построить трехмерные диаграммы рассеяния с помощью Matplotlib
Как изменить размер, непрозрачность и цвет точек данных в трех измерениях
Как добавить заголовки и метки осей к вашей трехмерной диаграмме рассеяния в Matplotlib

Оглавление

Создание 3D-диаграммы рассеяния в Matplotlib

Чтобы создавать трехмерные точечные диаграммы в Matplotlib, мы можем импортировать дополнительные вспомогательные модули из Matplotlib. Это позволяет нам легко проецировать данные на третье измерение. Давайте начнем с импорта наших библиотек и настройки некоторых переменных для построения графиков.

# Импорт библиотек и загрузка данных
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)  # Создание массива значений x от 0 до 25 с шагом 0.1
y = np.cos(x)  # Создание массива значений y как косинуса от x
z = np.sin(x)  # Создание массива значений z как синуса от x
print(x, y, z)

Давайте посмотрим, как мы можем визуализировать наши данные. Сначала мы настроим наш график, fig, и оси, ax, для отображения нашей визуализации. Наши оси укажут, что мы хотим проецировать данные в трех измерениях, передав projection='3d'.

Оттуда мы можем легко создать 3D-диаграмму рассеяния, используя функцию

# Создание трехмерной точечной диаграммы рассеяния в Matplotlib
fig = plt.figure()  # Создание новой фигуры
ax = plt.axes(projection='3d')  # Добавление трехмерных осей

ax.scatter3D(x, y, z)  # Построение трехмерной точечной диаграммы рассеяния
plt.show()  # Отображение графика

Это возвращает следующее изображение:

# Импорт библиотек и загрузка данных
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)  # Создание массива значений x от 0 до 25 с шагом 0.1
y = np.cos(x)  # Создание массива значений y как косинуса от x
z = np.sin(x)  # Создание массива значений z как синуса от x

# Создание трехмерной точечной диаграммы рассеяния в Matplotlib
fig = plt.figure()  # Создание новой фигуры
ax = plt.axes(projection='3d')  # Добавление трехмерных осей

ax.scatter3D(x, y, z)  # Построение трехмерной точечной диаграммы рассеяния
plt.show()  # Отображение графика

Давайте разберем, что мы сделали:

Мы импортировали наши библиотеки и создали некоторые переменные, содержащие данные
Мы затем создали наш рисунок и оси
Мы построили трехмерную диаграмму рассеяния на наших осях, передав три массива данных.
Наконец, мы показали график, используя plt.show().

Изменение цвета точек на 3D-диаграммах в Matplotlib

Добавление цвета к трехмерной визуализации может быть очень полезным. Трехмерное пространство бывает сложно воспринимать, и цвет помогает это упростить. Это позволяет нам лучше понять третье измерение.

Чтобы сделать это, мы можем использовать параметр c=. Это позволяет нам передать один цвет, если мы хотим использовать один и тот же цвет для всех точек, или массив чисел для окрашивания в зависимости от значения.

При работе с последовательными данными, как в нашем примере, можно использовать карты цветов. Это позволяет создать градиент для демонстрации изменений в данных. Различные карты цветов, предлагаемые библиотекой Matplotlib, можно найти здесь.

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

# Добавление цвета к нашей трехмерной точечной диаграмме рассеяния
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)
y = np.cos(x)
z = np.sin(x)

fig = plt.figure()
ax = plt.axes(projection='3d')

# Использование значения x в качестве цвета точек, заданных с помощью параметра c
ax.scatter3D(x, y, z, c=x, cmap='Blues')
plt.show()

Изменение размера точек на 3D-диаграммах в Matplotlib

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

Давайте используем переменную x для определения размера наших точек:

# Изменение размера маркера на трехмерной точечной диаграмме рассеяния
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)
y = np.cos(x)
z = np.sin(x)

fig = plt.figure()
ax = plt.axes(projection='3d')

# Использование значения x в качестве размера маркеров, заданного с помощью параметра s
ax.scatter3D(x, y, z, s=x)
plt.show()

Изменение непрозрачности точек на 3D-диаграммах в Matplotlib

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

Matplotlib позволяет настраивать прозрачность каждой точки, используя параметр alpha=, куда можно передать число с плавающей точкой. 0 обозначает полную прозрачность, в то время как 1 — отсутствие прозрачности.

Давай посмотрим, как это работает:

# Изменение прозрачности маркеров на трехмерной точечной диаграмме рассеяния
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)
y = np.cos(x)
z = np.sin(x)

fig = plt.figure()
ax = plt.axes(projection='3d')

# Использование параметра alpha для задания прозрачности маркеров
ax.scatter3D(x, y, z, alpha=0.7)
plt.show()

Добавление заголовков и меток осей к 3D-диаграммам рассеяния в Matplotlib

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

plt.title() установить заголовок
plt.set_xlabel() установить метку оси X
plt.set_ylabel()чтобы установить метку оси Y
ax.set_zlabel() чтобы установить метку оси Z

Давайте добавим к нашей диаграмме некоторые заголовки и подписи:

# Добавление заголовка и подписей к осям нашей трехмерной точечной диаграмме рассеяния
from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

x = np.arange(0, 25, 0.1)
y = np.cos(x)
z = np.sin(x)

fig = plt.figure()
ax = plt.axes(projection='3d')

# Добавление заголовка и подписей к осям
ax.scatter3D(x, y, z)
plt.title('Наша 3D-диаграмма рассеяния', size=18)
plt.xlabel('x-axis')
plt.ylabel('y-axis')
ax.set_zlabel('z-axis')
plt.show()

Заключение

В этом учебном пособии вы научились создавать 3D точечные диаграммы в Matplotlib. Вы научились использовать вспомогательные модули в Matplotlib для проецирования на третье измерение. Затем вы научились создавать 3D точечные диаграммы. После этого вы узнали, как изменять цвет, размер и прозрачность этих графиков. Наконец, вы научились добавлять заголовок и метки осей к вашему графику.

Дополнительные ресурсы

Чтобы узнать больше о схожих темах, ознакомьтесь с приведенными ниже руководствами:

Построение графиков на Python с помощью Matplotlib
Диаграммы рассеяния Matplotlib – узнайте все, что вам нужно знать
Линейные диаграммы Matplotlib – узнайте все, что вам нужно знать

PreviousКак построить функцию в Python с использованием Matplotlib NextКак изменить размер шрифта в графике Matplotlib

Last updated 4 months ago