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

Question

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

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

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'

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

Answer 1

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