动画3D随机游走

动画3D随机游走

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. import mpl_toolkits.mplot3d.axes3d as p3
  4. import matplotlib.animation as animation
  6. # Fixing random state for reproducibility
  7. np.random.seed(19680801)
  10. def Gen_RandLine(length, dims=2):
  11.     """
  12.     Create a line using a random walk algorithm
  14.     length is the number of points for the line.
  15.     dims is the number of dimensions the line has.
  16.     """
  17.     lineData = np.empty((dims, length))
  18.     lineData[:, 0] = np.random.rand(dims)
  19.     for index in range(1, length):
  20.         # scaling the random numbers by 0.1 so
  21.         # movement is small compared to position.
  22.         # subtraction by 0.5 is to change the range to [-0.5, 0.5]
  23.         # to allow a line to move backwards.
  24.         step = ((np.random.rand(dims) - 0.5) * 0.1)
  25.         lineData[:, index] = lineData[:, index - 1] + step
  27.     return lineData
  30. def update_lines(num, dataLines, lines):
  31.     for line, data in zip(lines, dataLines):
  32.         # NOTE: there is no .set_data() for 3 dim data...
  33.         line.set_data(data[0:2, :num])
  34.         line.set_3d_properties(data[2, :num])
  35.     return lines
  37. # Attaching 3D axis to the figure
  38. fig = plt.figure()
  39. ax = p3.Axes3D(fig)
  41. # Fifty lines of random 3-D lines
  42. data = [Gen_RandLine(25, 3) for index in range(50)]
  44. # Creating fifty line objects.
  45. # NOTE: Can't pass empty arrays into 3d version of plot()
  46. lines = [ax.plot(dat[0, 0:1], dat[1, 0:1], dat[2, 0:1])[0] for dat in data]
  48. # Setting the axes properties
  49. ax.set_xlim3d([0.0, 1.0])
  50. ax.set_xlabel('X')
  52. ax.set_ylim3d([0.0, 1.0])
  53. ax.set_ylabel('Y')
  55. ax.set_zlim3d([0.0, 1.0])
  56. ax.set_zlabel('Z')
  58. ax.set_title('3D Test')
  60. # Creating the Animation object
  61. line_ani = animation.FuncAnimation(fig, update_lines, 25, fargs=(data, lines),
  62.                                    interval=50, blit=False)
  64. plt.show()

下载这个示例