在运行Python脚本后保持Mayavi窗口开启
我正在尝试运行一个使用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()
这个图看起来很不错,但它马上就消失了。我看到说需要加上以下代码才能让图保持显示:
import mayavi
v = mayavi.mayavi()
v.master.wait_window()
这听起来挺合理的,但当我尝试这样做时,python却报告了:
AttributeError: 'module' object has no attribute 'mayavi'
有什么想法吗?我在OS X上使用的是标准的Enthought Python。