3D контурный график на Python

1. Импортируем библиотеки:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

> numpy — для работы с массивами
> matplotlib.pyplot — для построения графиков
> mpl_toolkits.mplot3d — модуль для 3D-графиков

2. Создаём координатную сетку:

x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)

🔹linspace создаёт равномерные значения по X и Y
🔹meshgrid строит двумерную сетку координат

3. Определяем функцию:

def f(x, y):
    return np.sin(np.sqrt(x**2 + y**2))

Это функция поверхности, значения которой будут отображаться через контуры по оси Z

4. Вычисляем значения Z:

Z = f(X, Y)

Получаем двумерный массив значений Z для каждой точки (X, Y)

5. Создаём фигуру и 3D-подграфик:

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

Создаём фигуру и добавляем 3D‑ось

6. Рисуем контуры в 3D:

contour = ax.contour3D(X, Y, Z, 50, cmap='viridis')

> contour3D рисует линии уровня
> 50 — число уровней
> cmap='viridis' — цветовая палитра

7. Подписываем оси и добавляем шкалу:

ax.set_xlabel('X-axis')
ax.set_ylabel('Y-axis')
ax.set_zlabel('Z-axis')
fig.colorbar(contour, ax=ax, label='Z values')

Добавляем подписи и цветовую шкалу (colorbar) для значений Z

8. Показываем график:

plt.show()

👉 @PythonPortal