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

时间:2013-05-12 12:14:06

标签: python matplotlib mayavi mplot3d mayavi.mlab

你能举例说明

吗?
  1. 最初画一个表面
  2. 然后在循环中绘制一些随机线。在绘制每一行之间,浪费了一点时间,给人一种动画的印象。
  3. 代码应该来自ipythonpydev

1 个答案:

答案 0 :(得分: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()