Python:在循环中连续添加到 3d 场景

Question

你能给出示例代码吗

1. 首先绘制一个曲面
2. 然后在循环中绘制一些随机线。在绘制每条线之间，会浪费一点时间，以给人动画的印象。

该代码应该可以在 ipython 和 pydev 上运行。

Answer 1

使用 mayavi 的一个答案如下:

import numpy as np
from mayavi import mlab
import time
from tvtk.tools import visual


# # prepare surface data
rng = 20
step = 1
X = np.arange(0, rng, step)
Y = X
Z = np.random.uniform(-1, 0, (rng, rng))

# # draw the surface
fig = mlab.figure(size=(500, 500), bgcolor=(1, 1, 1))
visual.set_viewer(fig)
s = mlab.surf(X, Y, Z)
mlab.axes(color=(0, 0, 0))
mlab.view(40, 40)
mlab.outline()

for i in xrange(5):
    # # sleep a little to give the impression of animation
    time.sleep(1)
    # # get coordinates of two random points
    p1 = np.random.uniform(0, 10, (3,)) 
    p2 = np.random.uniform(0, 10, (3,))
    line = np.vstack((p1, p2))
    x, y, z = (line[:, dim] for dim in xrange(3))

    # # connect points
    mlab.plot3d(x, y, z, figure=fig, tube_radius=.05, colormap='Greens')

mlab.show()