Я написал пакет для включения поддержки латекса для mayavi под названием mlabtex: https://github.com/MuellerSeb/mlabtex
. Он создает изображение, отображаемое с помощью matplotlib, и использует его в качестве текстуры для mlab.surf.Интерфейс похож на mlab.text3d.
С этим вы можете сделать что-то вроде этого:
import os
os.environ['QT_API'] = 'pyqt'
os.environ['ETS_TOOLKIT'] = 'qt4'
from mayavi import mlab
from mlabtex import mlabtex
TEXT = (r'Sebastian M\"uller, ' +
r'$f(x)=\displaystyle\sum_{n=0}^\infty ' +
r'f^{(n)}(x_0)\cdot\frac{(x-x_0)^n}{n!}$')
tex = mlabtex(0., 0., 0.,
TEXT,
color=(0., 0., 0.),
orientation=(30., 0., 0.),
dpi=1200)
mlab.axes()
mlab.show()
Чтобы пометить оси, вы можете теперь разместить текст там вручную.
Удачи!