大家晚上好,我是 Matplotlib 的长期用户,我最近发现了 Mayavi。
使用 Matplotlib,我可以绘制一个 3D 表面,并为每个轴绘制表面图的投影轮廓,我想知道 Mayavi 是否可以这样做。
这是迄今为止我使用 Matplotlib ( source ) 所做的一个示例,但我无法在 Internet 上找到使用 Mayavi 绘制轮廓的类似方法:
知道 Mayavi 的人可以告诉我是否可以像使用 Matplotlib 一样为每个轴绘制 3D 轮廓?
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()
玛雅维代码
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()