制作Mayavi Points3D绘图动画

Question

我正在尝试制作粒子轨迹的视频。但是，我的场景以某种方式从未更新过。这是一个非常简单的示例：

from __future__ import absolute_import, division, print_function
from mayavi import mlab
import numpy as np
import math

alpha = np.linspace(0, 2*math.pi, 100)

xs = np.cos(alpha)
ys = np.sin(alpha)
zs = np.zeros_like(xs)

mlab.points3d(0,0,0)
plt = mlab.points3d(xs[:1], ys[:1], zs[:1])

@mlab.animate(delay=100)
def anim():
    f = mlab.gcf()
    while True:
        for (x, y, z) in zip(xs, ys, zs):
            print('Updating scene...')
            plt.mlab_source.x[0] = x
            plt.mlab_source.y[0] = y
            plt.mlab_source.z[0] = z
            f.scene.render()
            yield


anim()
mlab.show()

如果运行此脚本，它将显示一个包含两个点和动画GUI的窗口。它还在终端上打印连续的“正在更新场景…”消息流。但是，场景根本不显示任何运动。

我究竟做错了什么？

Python 2.7，Mayavi 4.1，VTK 5.8

Answer 1

只需更改为：

…

    for (x, y, z) in zip(xs, ys, zs):
        print('Updating scene...')
        plt.mlab_source.set(x=x, y=y, z=z)
        yield

…

您甚至不需要f.scene.render()，根据说明文件，可以 mlab_source.set保证刷新。

另外，由于shape您的数据不会更改，因此您无需使用mlab_source.reset。

我也测试过并且工作正常。