如何使用Python极坐标绘制轨道天体位置并解决绘图异常问题

问题根因 #

• plt.polar(r, Th)报错的核心原因是参数顺序错误：matplotlib的plt.polar()要求第一个入参为极角（单位为弧度），第二个入参为极径，参数颠倒必然输出混乱图像。
• plt.polar(Th, r)参数顺序正确但修改偏心率后图形异常，大概率是两个原因：要么极角Theta的生成逻辑/单位错误，要么开普勒轨道的极径计算公式有误，也可能是偏心率取值超出了椭圆轨道的范围。

正确实现方法 #

核心公式说明 #

焦点在原点的开普勒椭圆轨道极坐标公式为：

$r = \frac{a(1-e^2)}{1 + e\cdot cos\theta}$
其中a为半长轴，e为偏心率，取值必须满足$0\leq e<1$（e=0为正圆，e≥1时轨道为抛物线/双曲线，不再是椭圆），$\theta$为极角，即你定义的焦点与天体连线和水平线的夹角。

代码示例 #

import numpy as np
import matplotlib.pyplot as plt

# 自定义轨道参数
a = 1.5 # 半长轴
e = 0.7 # 偏心率，可在0≤e<1范围内调整

# 生成极角序列：0到2π均匀采样，1000个点保证曲线平滑，单位为弧度
theta = np.linspace(0, 2 * np.pi, 1000)
# 按开普勒公式计算极径r
r = a * (1 - e ** 2) / (1 + e * np.cos(theta))

# 绘制极坐标图
plt.figure(figsize=(6,6))
ax = plt.subplot(111, projection='polar')
ax.plot(theta, r)
# 显式设置0度指向右侧（匹配你定义的水平线夹角规则，默认也是该配置）
ax.set_theta_zero_location('E')
# 显式设置角度逆时针递增（默认也是该配置）
ax.set_theta_direction(-1)
plt.show()

异常情况排查点 #

• 如果你生成的Theta是角度单位，必须先转弧度再传入绘图：theta_rad = np.deg2rad(Theta)
• 如果你是通过求解开普勒方程（$M = E - e\cdot sinE$，E为偏近点角）来计算r和θ，要注意转换逻辑是否正确：$cos\theta = \frac{cosE - e}{1 - e\cdot cosE}$，$r = a(1 - e\cdot cosE)$，转换错误也会导致图形异常。
• 偏心率e不要设置为≥1的值，此时轨道本身就不是椭圆。

内容的提问来源于stack exchange，提问作者BHavIt