我想在星球动画中有踪迹

Question

我一直在研究 2 体问题，其中涉及行星围绕恒星的动画。我希望我的星球在移动 this 之类的东西时留下一点痕迹。除了这一件事，我的代码完美地绘制了所有内容。获取不到星球的踪迹

fig, ax = plt.subplots()

print(func(0.98, -np.pi, 0, 0, 0.001))
ax.set(xlim = (-1.2,1.5), ylim = (-1.5,1.5), xlabel = 'x axis', ylabel='y axis', title = 'Planet\'s orbit')

def polar_animator(i):
    trail = 40
    l1.set_data(x_graph[i-1:i], y_graph[i-1:i])
    return l1,

l1, = ax.plot([],[], 'o-')
l2, = ax.plot([-0.8],[0],marker= 'o')

    
ani = animation.FuncAnimation(fig, polar_animator, frames= len(x_graph), interval=5, blit=True)
ani.save('planet.mp4', writer= 'ffmpeg')

我得到的输出只是一个绕太阳运动的球。

Answer 1

我认为你非常接近。我所做的修改是：

保留最后 trail 点以显示在图表上（您只显示最后 2 个点）
在 plot() 调用中使用 markevery=[-1] 以仅在行尾显示标记。

完整代码：

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np

x_graph = np.linspace(0, 1, 100)
y_graph = x_graph
trail = 40

fig, ax = plt.subplots()
ax.set(xlim=(-1.2, 1.5), ylim=(-1.5, 1.5), xlabel='x axis', ylabel='y axis', title='Planet\'s orbit')


def polar_animator(i, trail=40):
    l1.set_data(x_graph[i - trail:i], y_graph[i - trail:i])
    return l1,


l1, = ax.plot([], [], 'o-', markevery=[-1])
l2, = ax.plot([-0.8], [0], marker='o')

ani = animation.FuncAnimation(fig, polar_animator, frames=len(x_graph), fargs=(trail,), interval=5, blit=True)

我想在星球动画中有踪迹

I want to have a trail in the planet animation

python

arrays

animation

matplotlib

matplotlib-animation