На этом шаге мы рассмотрим пример создания такого приложения.
Создайте в вашем текстовом редакторе документ с именем pr152_1.py, в котором мы напишем код, демонстрирующий работу с библиотекой Mayavi.
Для начала импортируем нужные библиотеки:
import numpy as np from mayavi import mlab
С numpy вы познакомились на предыдущих шагах. Из библиотеки Mayavi мы импортируем модуль mlab. Он предоставляет инструменты для построения 2D/3D-моделей в Matplotlib-стиле. Вы можете использовать mlab как в коде своих собственных модулей, так и в интерактивной среде.
Построим набор точек от -5 до 5 в количестве 100 штук, для этого воспользуемся функцией linspace() из numpy:
t = np.linspace(-5, 5, 100)
Создадим массивы с координатами точек х, y, z для трёхмерной спирали через параметрические уравнения:
x = 3 * np.cos(t) y = 3 * np.sin(t) z = t / np.pi
Добавим код для построения 3D-изображения:
s = mlab.plot3d(x, y, z, z, tube_radius=0.1) mlab.show()
Приведём весь код, который должен быть в вашем файле:
import numpy as np from mayavi import mlab t = np.linspace(-5, 5, 100) x = 3 * np.cos(t) y = 3 * np.sin(t) z = t / np.pi s = mlab.plot3d(x, y, z, z, tube_radius=0.1) mlab.show()
Запустите приложение. В результате должно открыться окно "Mayavi Scene" с моделью, представленной на рисунке 1.
Рис.1. Пример построения 3D-модели в Python с использованием Mayavi
Mayavi Scene является интерактивной средой, в ней вы можете вращать изображения, приводить к изометрическому виду, сохранять и настраивать его.
Со следующего шага мы начнем рассматривать настройку представления.