运行 python 脚本后在 Mayavi 中保留窗口

Question

我正在尝试运行一个使用 Mayavi 生成绘图的 python 脚本。

我正在使用以下示例代码来生成等值面图：

from mayavi import mlab
def test_contour3d():
    x, y, z = numpy.ogrid[-5:5:64j, -5:5:64j, -5:5:64j]
    scalars = x * x * 0.5 + y * y + z * z * 2.0
    obj = mlab.contour3d(scalars, contours=4, transparent=True)
    return obj

test_contour3d()

情节看起来很棒，但它立即消失了。我read，您需要包含以下代码才能使情节持续下去：

import mayavi
v = mayavi.mayavi()
v.master.wait_window()

这似乎很合理，但是当我尝试这个时，python 报告：

AttributeError: 'module' object has no attribute 'mayavi'

有什么想法吗？我在 OS X 上使用标准的 Enthought Python。

Answer 1

使用mlab.show() 保持图框打开。