Mayavi: поверхностный сюжет и трехмерные контуры - PullRequest
1 голос
/ 01 июня 2019

Добрый вечер всем, я давно пользуюсь Matplotlib, и недавно я обнаружил Mayavi.

С Matplotlib я могу построить трехмерную поверхность с спроецированными контурами графика поверхности для каждой оси, и мне было интересноесли то же самое можно сделать с Mayavi.

Вот пример того, что я сделал до сих пор с Matplotlib ( source ), но я не смог найти аналогичный способпостроение контуров с Mayavi в Интернете:

enter image description here

Может ли кто-нибудь, кто знает Mayavi, сказать мне, могу ли я построить 3D контуры для каждой оси, как сMatplotlib?


Код Matplotlib

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
from matplotlib import cm

fig = plt.figure()
ax = fig.gca(projection='3d')
X, Y, Z = axes3d.get_test_data(0.05)
ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=0.3)
cset = ax.contour(X, Y, Z, zdir='z', offset=-100, cmap=cm.coolwarm)
cset = ax.contour(X, Y, Z, zdir='x', offset=-40, cmap=cm.coolwarm)
cset = ax.contour(X, Y, Z, zdir='y', offset=40, cmap=cm.coolwarm)

ax.set_xlabel('X')
ax.set_xlim(-40, 40)
ax.set_ylabel('Y')
ax.set_ylim(-40, 40)
ax.set_zlabel('Z')
ax.set_zlim(-100, 100)

plt.show()

Код Mayavi

from mpl_toolkits.mplot3d import axes3d
from mayavi import mlab

X, Y, Z = axes3d.get_test_data(0.05)

Z = np.rollaxis(Z,0,2)
X = np.rollaxis(X,0,2)
Y = np.rollaxis(Y,0,2)

mlab.surf(X, Y, Z, warp_scale="auto", opacity=1)
mlab.axes(xlabel='X', ylabel='Y', zlabel='Z')
mlab.show()
...