аннотирование многих точек текста в майави с использованием mlab

Я пытаюсь аннотировать точки, построенные с помощью функции points3d (), используя mayavi.mlab. Каждая точка связана с меткой, которую я хотел бы построить рядом с точками, используя функцию text3d (). Построение точек происходит быстро, однако функция mlab.text3d (), похоже, не принимает массивы координат, поэтому мне приходится перебирать точки и печатать текст индивидуально, что очень медленно:

for i in xrange(0, self.n_labels): self.mlab_data.append( mlab.points3d( pX[self.labels == self.u_labels[i], 0], pX[self.labels == self.u_labels[i], 1], pX[self.labels == self.u_labels[i], 2], color=self.colours[i], opacity=1, scale_mode="none", scale_factor=sf ) ) idcs, = np.where(self.labels == self.u_labels[i]) for n in idcs.flatten(): mlab.text3d( pX[n, 0], pX[n, 1], pX[n, 2], "%d" % self.u_labels[i], color=self.colours[i], opacity=1, scale=sf ) 

Любые идеи, как я мог бы ускорить это? Кроме того, можно добавить легенду (как, например, в matplotlib), я не смог найти ничего в документах.

• Работает ли свойство текста шрифта Mayavi «Размер шрифта»?
• Добавление 4-й переменной в 3D-график в Python
• Измените источники данных mlab quiver3d & surf без очистки фигуры в скриптах признаков
• Как напрямую установить цвета RGB / RGBA в майави
• Mayavi: интерполировать цвета лица в triangular_mesh

Благодаря,

Патрик

  import scipy from mayavi import mlab X = 100 * scipy.rand(100, 3) figure = mlab.figure('myfig') figure.scene.disable_render = True # Super duper trick mlab.points3d(X[:,0], X[:,1], X[:,2], scale_factor=0.4) for i, x in enumerate(X): mlab.text3d(x[0], x[1], x[2], str(i), scale=(2, 2, 2)) figure.scene.disable_render = False # Super duper trick