更改Mayavi动画中的点数

Question

这是Python Mayavi中的简单动画脚本：

from mayavi import mlab
import numpy as np
alpha = np.linspace(0, 2*np.pi, 100)
beta = np.linspace(0, np.pi, 100)
x = np.sin(beta) * np.cos(alpha)
y = np.sin(beta) * np.sin(alpha)
z = np.cos(beta)
plt = mlab.points3d(x, y, z)
@mlab.animate(delay=100)
def anim():
    global x, y, z
    f = mlab.gcf()
    for _ in range(100):
        # x = np.concatenate((x, [np.random.random()]))
        # y = np.concatenate((y, [np.random.random()]))
        # z = np.concatenate((z, [np.random.random()]))
        x = 1.1 * x
        plt.mlab_source.set(x=x, y=y, z=z)
        f.scene.render()
        yield

anim()
mlab.show()

这运行良好，积分四处移动。但是，我想取消注释np.concatenate行，以使动画过程中的点数变化... Mayavi似乎不支持这一点？

我认为这种限制与更新情节的效率有关，但是我希望上述工作并不介意任何速度打击。

有什么想法？

我尝试简单地在mlab.clf()之后简单地重新启动mlab.points3d(x, y, z)，但是动画不显示 - 只有最后一个帧。

预先感谢您。

Answer 1

您应该根据文档使用reset()而不是set()：

x, y = np.mgrid[0:3:1,0:3:1]
s = mlab.surf(x, y, np.asarray(x*0.1, 'd'),
            representation='wireframe')
# Animate the data.
fig = mlab.gcf()
ms = s.mlab_source
for i in range(5):
    x, y = np.mgrid[0:3:1.0/(i+2),0:3:1.0/(i+2)]
    sc = np.asarray(x*x*0.05*(i+1), 'd')
    ms.reset(x=x, y=y, scalars=sc)
    fig.scene.reset_zoom()