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

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

我使用以下示例代码生成iso曲面图：

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()

情节看起来很棒，但马上就消失了。我建议您需要包含以下代码以使绘图保持不变：

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

这似乎很合理，但当我尝试此方法时，python报告：

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

---

有什么想法吗？我在OSX上使用的是标准的Enthound Python。

用于保持绘图框架打开

这很容易：）