Mayavi, как показать сетку осей - PullRequest
Новая
       76

Mayavi, как показать сетку осей

0 голосов
/ 25 февраля 2019

В Mayavi я хотел бы видеть сетку осей на следующем графике 

# Source: <<https://scicomp.stackexchange.com/a/23148/10048>>.
import numpy as np
from mayavi import mlab

# Test data: Matlab `peaks()`
x, y = np.mgrid[-3:3:50j, -3:3:50j]
z = 3*(1 - x)**2 * np.exp(-x**2 - (y + 1)**2) \
   - 10*(x/5 - x**3 - y**5)*np.exp(-x**2 - y**2) \
   - 1./3*np.exp(-(x + 1)**2 - y**2) 

mlab.figure(bgcolor=(1, 1, 1))  # Make background white.
surf = mlab.surf(x, y, z, colormap='RdYlBu', warp_scale=0.3, representation='wireframe', line_width=0.5)
mlab.outline(color=(0, 0, 0))
axes = mlab.axes(color=(0, 0, 0), nb_labels=5)
axes.title_text_property.color = (0.0, 0.0, 0.0)
axes.title_text_property.font_family = 'times'
axes.label_text_property.color = (0.0, 0.0, 0.0)
axes.label_text_property.font_family = 'times'
# mlab.savefig("vector_plot_in_3d.pdf")
mlab.gcf().scene.parallel_projection = True  # Source: <<https://stackoverflow.com/a/32531283/2729627>>.
mlab.orientation_axes()  # Source: <<https://stackoverflow.com/a/26036154/2729627>>.
mlab.show()

как в Matlab

The axes grid is shown as I want here

Какую команду я должен использовать в Майави для достижения этой цели?

1 Ответ

0 голосов
/ 25 февраля 2019

Мое предложение будет использовать пользовательский фильтр, например CubeAxesActor из VTK (который кажется более полным, чем фильтр Axes в Mayavi по умолчанию).Который в Mayavi переведет что-то вроде 

...
from mayavi import mlab
from tvtk.api import tvtk

...
# Load data
...
mlab.pipeline.user_defined(data, filter=tvtk.CubeAxesActor())
...

Но, похоже, Mayavi не нравится этот фильтр 

traits.trait_errors.TraitError: Cannot set the undefined 'input_connection' attribute of a 'CubeAxesActor' object.

Я не уверен, связан ли он с моей версией VTK, но выможет попробовать

...