python的matplotlib的模擬太陽-地球-月亮運動


#---第1步---導出模塊---
import numpy as np
import matplotlib as mpl
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animmation
#導出通用字體設置
from matplotlib import font_manager 
#定義引出字體模塊、位置、大小
my_font = font_manager.FontProperties(fname="hwfs.ttf",size=20)

#---第2步---初始化定義---
#r1的大小與月球的速度和距離地球距離有關,越大越不好,建議10
r1 = 10
#r2是月球與地球的半徑大小
r2 = 2
#π=圓周率(Pai)是圓的周長與直徑的比值
#omega1=2π是一個圓,是地球的運動一圈;1×π=為半圓。
omega1 = 2 * np.pi
#定義omega2為幾個π,與月球的公轉速度有關
#建議24π=代表月球以地球公轉的一圈分12部分,12+12半圈
#比如48π=24+24半圈,即將地球繞太陽一圈在分24部分,月球自轉和公轉的速度也加快
omega2 = 48 * np.pi
#月球自轉和公轉的角度與地球公轉的水平夾角
phi = 5 * np.pi / 180 

#---第3步---更新函數定義---
def update(data):
    #聲明為全局變量
    global line1, line2 , line3
    #地球公轉運動的更新
    line1.set_data([data[0], data[1]])
    line1.set_3d_properties(data[2])
    #月球運動更新
    line2.set_data([data[3], data[4]])
    line2.set_3d_properties(data[5])
    #月球自轉線的更新
    line3.set_data([data[6], data[7]])
    line3.set_3d_properties(data[8])
    return line1,line2,line3,

#---第4步---初始化框架---
def init():
    global line1, line2, line3
    ti = 0
    t = t_drange[np.mod(ti, t_dlen)]
    xt1 = x0 + 2*r1 * np.cos(omega1 * t)
    yt1 = y0 + 2*r1 * np.sin(omega1 * t)
    zt1 = z0 + 0
    xt2 = xt1 + 2*r2 * np.sin(omega2 * t)
    yt2 = yt1 + 2*r2 * np.cos(omega2 * t)/(np.cos(phi) * (1 + np.tan(phi) ** 2))
    zt2 = zt1 + (yt2 - yt1) * np.tan(phi)
    xt21 = xt1 + r2 * np.sin(2 * np.pi * t_range)
    yt21 = yt1 + r2 * np.cos(2 * np.pi * t_range)/(np.cos(phi) * (1 + np.tan(phi) ** 2))
    zt21 = zt1 + (yt21 - yt1) * np.tan(phi)

    #地球位置、形狀、顏色、大小設置
    line1, = ax.plot([xt1], [yt1], [zt1], marker='o', color='blue',markersize=20)
    #月球位置、形狀、顏色、大小設置
    line2, = ax.plot([xt2], [yt2], [zt2], marker='o', color='orange',markersize=12)
    #月球繞地球的軌跡線和顏色purple=紫色
    line3, = ax.plot(xt21, yt21, zt21, color='purple')
    return line1,line2,line3,

#---第5步---運動數據的產生---
def data_gen():
    global x0,y0,z0,t_dlen
    data = []
    for ti in range(1,t_dlen):
        t = t_drange[ti]
        xt1 = x0 + r1 * np.cos(omega1 * t)
        yt1 = y0 + r1 * np.sin(omega1 * t)
        zt1 = z0
        xt2 = xt1 + r2 * np.sin(omega2 * t)
        yt2 = yt1 + r2 * np.cos(omega2 * t)/(np.cos(phi) * (1 + np.tan(phi) ** 2))
        zt2 = zt1 + (yt2 - yt1) * np.tan(phi)
        xt21 = xt1 + r2 * np.sin(2 * np.pi * t_range)
        yt21 = yt1 + r2 * np.cos(2 * np.pi * t_range)/(np.cos(phi) * (1 + np.tan(phi) ** 2))
        zt21 = zt1 + (yt21 - yt1) * np.tan(phi)
        data.append([xt1, yt1, zt1, xt2, yt2, zt2, xt21, yt21, zt21])
    return data

#---第6步---定義取值范圍0~10,每個0.005取
#地球公轉的軌跡線刻度,越小越好,建議0.005,否則軌跡線不是圓形,有鋸齒樣類圓形
t_range = np.arange(0, 10 + 0.005, 0.005)
#地球公轉速度,越大速度越大,建議0.005,
t_drange = np.arange(0,10, 0.005 )
t_len = len(t_range)
t_dlen = len(t_drange)

#---第7步---三大星球的大小、顏色、坐標---
#太陽的坐標位置,三維坐標
x0 =y0=z0= 0

#地球的運動中的坐標
x1 = x0  + r1 * np.cos(omega1 * t_range)
y1 = y0 + r1 * np.sin(omega1 * t_range)
z1 = z0 + np.zeros(t_len)

#---第8步---定義圖片f和ax等---
#窗口大小也就是展示圖片的畫布大小:22=2200,14=1400,即2200×1400
#這是窗口的背景顏色,有區別,默認白色
f = plt.figure(figsize=(22,14),facecolor='black',edgecolor='white')
#這是畫布的背景顏色,默認白色
ax  = f.add_subplot(111,projection='3d',facecolor='black')

#---第9步---太陽和地球軌跡線設置---
#太陽的設置顏色,位置,大小
ax.plot([0], [0], [0], marker='o', color= 'red', markersize=100)
#地球公轉的軌跡線和顏色g=green=綠色,三維坐標
ax.plot(x1, y1, z1, 'g')

#---第10步---圖片的坐標刻度設置---
#坐標軸刻度雖然不顯示,但刻度的標記對整個圖形有一定的拉伸影響
#x坐標軸刻度范圍
ax.set_xlim([-(r1 + 2), (r1 + 2)])
#y坐標軸刻度范圍
ax.set_ylim([-(r1 + 5), (r1 + 5)])
#z坐標軸刻度范圍
ax.set_zlim([-15, 15])

#---第11步---圖片標題等設置---
#圖示的標題
#動畫走起,f圖片掛起動畫里,不斷更新,interval = 20=數值越小,速度越快
ani = animmation.FuncAnimation(f, update, frames = data_gen(), init_func = init,interval = 20)
#坐標及其刻度隱藏
plt.axis('off')
#圖片標題、字體、顏色
plt.title(u'太陽-地球-月亮模擬示意圖', fontproperties=my_font,color='r')
#圖片展現
plt.show()

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM