Matplotlib绘制3D行星轨道出现多余线条问题求助

多余棕色线条的原因 #

你遇到的3D模式下的多余棕色线条，是因为matplotlib的ax.plot()在3D模式下会将连续传入的多组坐标首尾相连。你现在把多个行星的x、y、z数组一次性传给plot()，它会把前一个行星轨道的最后一个点和下一个行星轨道的第一个点用线条连起来，这些就是你看到的多余线条。

修复后的代码 #

把一次性传入多组数据的写法改成循环逐个绘制每个行星轨道，就能解决这个问题：

import matplotlib.pyplot as plt
import numpy as np

Mass = [332837, 0.055, 0.85, 1.000, 0.107, 317.85, 95.159, 14.500, 17.204, 0.003]
a = ["", 0.387, 0.723, 1.000, 1.523, 5.202, 9.576, 19.293, 30.246, 39.509]
e = ["", 0.21, 0.01, 0.02, 0.09, 0.05, 0.06, 0.05, 0.01, 0.25]
beta = ["", 7.00, 3.39, 0.00, 1.85, 1.31, 2.49, 0.77, 1.77, 17.5]

theta = np.linspace(0, 2*np.pi, 1000)


def get_planet_data(index, projection="2d"):
    r = (a[index] * (1 - e[index]**2))/ (1 - e[index] * np.cos(theta))
    x = r*np.cos(theta)
    y = r*np.sin(theta)
    if projection == "3d":
        angle = beta[index] * np.pi / 180
        xx = x*np.cos(angle)
        yy = y
        zz = x*np.sin(angle)
        return xx, yy, zz
    return x, y


fig = plt.figure()
ax = plt.axes(projection='3d')
ax.set_xlim(-40, 40)
ax.set_ylim(-40, 40)
ax.set_zlim(-40, 40)

# 循环逐个绘制每个行星轨道，避免轨道间的多余连线
for i in range(1, 10):
    xx, yy, zz = get_planet_data(i, "3d")
    ax.plot(xx, yy, zz)

plt.show()

关于3D绘图替代方案 #

你提到matplotlib的3D绘图能力有限，pyqtgraph和Mayavi确实是更合适的选择：

• pyqtgraph：基于Qt框架，交互响应快，性能优于matplotlib 3D，适合实时数据可视化场景
• Mayavi：专注于科学数据的3D可视化，功能更全面，支持复杂的3D渲染与交互操作

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