动画极坐标图

Question

我正在尝试使用 python 制作行星运动的动画图。当我没有动画时，我可以获得正确的路径显示，但是一旦我尝试对其进行动画处理，它就会显示为空白，即使当我尝试在控制台中打印每个值时， r 和 i 被输出为正确的值。

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

fig = plt.figure(figsize=(6,6))
ax = plt.subplot(111, polar=True)
ax.set_ylim(0,1)
line, = ax.plot([],[])

semi_major_axis = 1
eccentricity = 0.1
theta = np.linspace(0,2*np.pi, num = 50)

point, = ax.plot(0,1, marker="o")

def frame(i):
    r=(semi_major_axis*(1-eccentricity**2)/(1-eccentricity*np.cos(i)))
    line.set_xdata(i)
    line.set_ydata(r)
    return line,


ax.set_rmax(semi_major_axis+1)
ax.set_rticks(np.linspace(0,semi_major_axis+1, num = 5))
ax.set_rlabel_position(-22.5) 


animation = FuncAnimation(fig, func=frame, frames=theta, interval=10)
plt.show()

Answer 1

我认为问题是你正在绘制一个点而不是一条线。以下代码对我有用：

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

    fig = plt.figure(figsize=(6, 6))
    ax = plt.subplot(111, polar=True)
    ax.set_ylim(0, 1)

    semi_major_axis = 1
    eccentricity = 0.1
    theta = np.linspace(0, 2 * np.pi, num=50)

    def frame(i):
        r = (semi_major_axis * (1 - eccentricity ** 2) / (1 - eccentricity * np.cos(i)))
        ax.plot(i,r,'b', marker='o')

    ax.set_rmax(semi_major_axis + 1)
    ax.set_rticks(np.linspace(0, semi_major_axis + 1, num=5))
    ax.set_rlabel_position(-22.5)

    animation = FuncAnimation(fig, func=frame, frames=theta, interval=10)
    plt.show()