运行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()保持绘图框打开。