动画散点图

Question

我想用 python 做一个动画，逐点表示每个点。我像往常一样这样做，但它不起作用。有什么帮助吗？ 我尝试了两种不同的方法。

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

%matplotlib nbagg

def mcd(a, b):
    resto = 0
    while(b > 0):
        resto = b
        b = a % b
        a = resto
    return a

N = 1200

n = list (range (N))
an = [1,1]

for i in range (2,N):
    k = i-1
    if mcd (n[i], an[k]) == 1:
        an.append (n[i] + 1 + an[k])
    else:
        an.append (an[k]/mcd (n[i], an[k]))

fig = plt.figure ()
ax = fig.add_subplot (111)
ax.grid (True)
ax.set_xlim(0, N*1.1)
pt, = ax.plot ([],[],'ko', markersize=2)

ax.plot (n,an, 'ko', markersize=2)

def init ():

    pt.set_data([],[])
    return (pt)

def animate (i,pt):

    pt.set_data (n[:i],an[:i])

    return (pt)

ani = FuncAnimation (fig, animate, fargs = (pt), frames=N, init_func=init, interval=50, blit = True)

plt.show ()

第二种方式：

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

%matplotlib nbagg

def mcd(a, b):
    resto = 0
    while(b > 0):
        resto = b
        b = a % b
        a = resto
    return a

N = 1200

n = list (range (N))
an = [1,1]

for i in range (2,N):
    k = i-1
    if mcd (n[i], an[k]) == 1:
        an.append (n[i] + 1 + an[k])
    else:
        an.append (an[k]/mcd (n[i], an[k]))

xdata, ydata = [],[]

fig = plt.figure ()
ax = fig.add_subplot(111)
ax.grid (True)
pt, = ax.plot ([],[],'ko', markersize=2)

ax.plot (n,an, 'ko', markersize=2)

def init ():
    ax.set_xlim(0, N*1.1)
    pt.set_data([],[])
    return (pt)

def animate (pt):
    xdata.append (n[i])
    ydata.append (an[i])

    pt.set_data (xdata,ydata)

    return (pt)

ani = FuncAnimation (fig, animate, fargs = (pt), frames=N, init_func=init, interval=50, blit = True)

plt.show ()

使用这些代码，我得到了包含所有点的整个图形。我想以动画的方式逐点填充图表。

Answer 1

以下将起作用

%matplotlib nbagg
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

def mcd(a, b):
    resto = 0
    while(b > 0):
        resto = b
        b = a % b
        a = resto
    return a

N = 1200

n = list (range (N))
an = [1,1]

for i in range (2,N):
    k = i-1
    if mcd (n[i], an[k]) == 1:
        an.append (n[i] + 1 + an[k])
    else:
        an.append (an[k]/mcd (n[i], an[k]))

fig = plt.figure ()
ax = fig.add_subplot (111)
ax.grid (True)
ax.set_xlim(0, N*1.1)
ax.set_ylim(min(an), max(an))
pt, = ax.plot([],[],'ko', markersize=2)

def init ():
    pt.set_data([], [])
    return pt,

def animate(i):
    pt.set_data (n[:i], an[:i])
    return pt,

ani = FuncAnimation (fig, animate, frames=N, init_func=init, interval=50, blit = True)

plt.show ()

Answer 2

编辑： 我在普通的 Python Shell 中测试了它，它在红点上绘制了黑点，但是 Jupyter 绘制了隐藏在红点后面的黑点，所以它需要这些线的顺序不同——首先是红点，接下来是黑点的空图。

ax.plot(n, an, 'ro', markersize=2) # red dots
pt, = ax.plot([], [], 'ko', markersize=2)

---

首先：我收到错误消息

 TypeError: 'Line2D' object is not iterable. 

都是因为

()

不创建元组 - 你必须使用逗号

,

来创建元组

return pt,

和

fargs=(pt,)

---

问题是因为你在开始使用时绘制了所有点

ax.plot(n, an, 'ko', markersize=2)

后来它在相同的地方画点，所以你看不到动画。

如果你使用不同的颜色 - 即。

red

ax.plot(n, an, 'ro', markersize=2)

然后你会看到红点上有黑点的动画。

或者删除这条线，它会在空窗口中画点。

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

#%matplotlib nbagg

def mcd(a, b):
    resto = 0
    while(b > 0):
        resto = b
        b = a % b
        a = resto
    return a

N = 1200

n = list(range(N))
an = [1, 1]

for i in range(2, N):
    k = i-1
    if mcd(n[i], an[k]) == 1:
        an.append(n[i] + 1 + an[k])
    else:
        an.append(an[k]/mcd(n[i], an[k]))

fig = plt.figure()
ax = fig.add_subplot(111)
ax.grid(True)
ax.set_xlim(0, N*1.1)
pt, = ax.plot([], [], 'ko', markersize=2)

ax.plot(n, an, 'ro', markersize=2) # red dots

def init():
    pt.set_data([], [])
    return (pt,)

def animate(i, pt):
    pt.set_data(n[:i], an[:i])
    return (pt,)


ani = FuncAnimation(fig, animate, fargs=(pt,), frames=N, init_func=init, interval=50, blit=True)

plt.show ()

---

在第二个代码中你有同样的问题，你也忘记了

i

在

def animate(i, pt):

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

#%matplotlib nbagg

def mcd(a, b):
    resto = 0
    while(b > 0):
        resto = b
        b = a % b
        a = resto
    return a

N = 1200

n = list(range (N))
an = [1, 1]

for i in range(2, N):
    k = i-1
    if mcd(n[i], an[k]) == 1:
        an.append(n[i] + 1 + an[k])
    else:
        an.append(an[k]/mcd (n[i], an[k]))

xdata, ydata = [], []

fig = plt.figure()
ax = fig.add_subplot(111)
ax.grid(True)
pt, = ax.plot([], [], 'ko', markersize=2)

ax.plot(n, an, 'ro', markersize=2)

def init():
    ax.set_xlim(0, N*1.1)
    pt.set_data([], [])
    return (pt,)

def animate(i, pt):
    xdata.append(n[i])
    ydata.append(an[i])

    pt.set_data(xdata, ydata)

    return (pt,)

ani = FuncAnimation(fig, animate, fargs=(pt,), frames=N, init_func=init, interval=50, blit=True)

plt.show ()