python太阳系行星动态演示

1. import numpy as np
2. import matplotlib as mpl
3. mpl.use(“TkAgg”)
4. from matplotlib import pyplot as plt
5. from mpl_toolkits.mplot3d import Axes3D#使用mpl_toolkits.mplot3d下Axes3D绘制3D图形
6. import matplotlib.animation as animmation
8. r1=5
9. r2=8
10. r3=11
11. r4=16
12. r5=20
13. r6=27
14. r7=36
15. r8=50#设置轨道半径
16. omega1 = 13* np.pi
17. omega2 = 9* np.pi
18. omega3 = 7* np.pi
19. omega4 = 6* np.pi
20. omega5 = 5* np.pi
21. omega6 = 4* np.pi
22. omega7 = 3* np.pi
23. omega8 = 2* np.pi#设置行星角速度
27. def update(data):#随时间t的变换设置新的行星位置坐标，也即是随着animmation.FuncAnimation函数的帧数参数变化每一帧的不同坐标位置的图片对象参数

28. global line1,line2,line3,line4,line5,line6,line7,line8  
    

29. line1.set_data([data[0], data[1]])  
    

30. line1.set_3d_properties(data[2])  
    

31. line2.set_data([data[3],data[4]])  
    

32. line2.set_3d_properties(data[5])  
    

33. line3.set_data([data[6], data[7]])  
    

34. line3.set_3d_properties(data[8])  
    

35. line4.set_data([data[9], data[10]])  
    

36. line4.set_3d_properties(data[11])  
    

37. line5.set_data([data[12], data[13]])  
    

38. line5.set_3d_properties(data[14])  
    

39. line6.set_data([data[15], data[16]])  
    

40. line6.set_3d_properties(data[17])  
    

41. line7.set_data([data[18], data[19]])  
    

42. line7.set_3d_properties(data[20])  
    

43. line8.set_data([data[21], data[22]])  
    

44. line8.set_3d_properties(data[23])  
    

45. return line1,line2,line3,line4,line5,line6,line7,line8  
    

46. def init():#行星的起始位置

47. global line1, line2, line3,line4,line5,line6,line7,line8