如何通过mayavi动画imshow

Question

我正在尝试使用mayavi动画imshow（）（即随着时间的推移更改图像），按照方法动画points3d（Animating a mayavi points3d plot），但场景永远不会更新。以下代码继续显示相同的（初始）图像，我将其设置为零。

非常感谢你的帮助！

注意：我知道如何在matploblib（matplotlib imshow(): how to animate?）中执行此操作。我现在想在mayavi做这件事。

我的环境是： Ubuntu 16.04 python 2.7.12 / ipython 4.2.0（都失败了）

    import numpy as np
    from mayavi import mlab

    @mlab.animate(delay=10)
    def main():
        s = np.zeros((2,2))
        im = mlab.imshow(s, colormap='gist_earth')
        for i in range(100):
            print i
            s += np.random.random((2,2))
            im.mlab_source.set(scalars=s)
            yield

    main()
    mlab.show()