地球轨道图Python

import numpy as np
import matplotlib.pyplot as plt
from astropy.coordinates import EarthLocation, get_body_barycentric_posvel
from astropy.time import Time
import astropy.units as u

# 设置时间范围
times = Time.now().jd + np.linspace(0, 365, 100)  # 一年内的时间点

# 获取地球和卫星的位置
earth_loc = EarthLocation.from_geodetic(0, 0)
satellite_posvel = get_body_barycentric_posvel('earth', times)

# 提取位置信息
x, y, z = satellite_posvel[0].xyz.to(u.km).value.T

# 绘制轨道图
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(111, projection='3d')
ax.plot(x, y, z, label='Satellite Orbit')
ax.scatter(0, 0, 0, color='red', label='Earth')  # 地球位置
ax.set_xlabel('X (km)')
ax.set_ylabel('Y (km)')
ax.set_zlabel('Z (km)')
ax.legend()
plt.show()