在matplotlib中使用for循环定义要绘制动画的多个绘图

时间:2014-02-21 15:07:22

标签: python animation matplotlib

Thanks to Jake Vanderplas,我知道如何开始使用matplotlib对动画情节进行编码。以下是示例代码:

from matplotlib import pyplot as plt
from matplotlib import animation

fig = plt.figure()

ax = plt.axes(xlim=(0, 2), ylim=(0, 100))

line, = plt.plot([], [])

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

def animate(i):
    line.set_data([0, 2], [0,i])
    return line,

anim = animation.FuncAnimation(fig, animate, init_func=init,
                               frames=100, interval=20, blit=True)

plt.show()

现在假设我想在循环的帮助下绘制大量函数(比如说四个)。我做了一些伏都教编程,试图理解如何模仿下面的逗号,这就是我得到的(不用说它不起作用:AttributeError: 'tuple' object has no attribute 'axes')。

from matplotlib import pyplot as plt
from matplotlib import animation

fig = plt.figure()

ax = plt.axes(xlim=(0, 2), ylim=(0, 100))

line = []
N = 4

for j in range(N):
    temp, = plt.plot([], [])
    line.append(temp)

line = tuple(line)

def init():
    for j in range(N):
        line[j].set_data([], [])
    return line,

def animate(i):
    for j in range(N):
        line[j].set_data([0, 2], [10 * j,i])
    return line,

anim = animation.FuncAnimation(fig, animate, init_func=init,
                               frames=100, interval=20, blit=True)

plt.show()

我的一些问题是:我该如何让它发挥作用?奖金(可能已关联):line, = plt.plot([], [])line = plt.plot([], [])之间有什么区别?

由于

1 个答案:

答案 0 :(得分:15)

在下面的解决方案中,我展示了一个更大的例子(还有条形图),可以帮助人们更好地理解应该为其他案例做些什么。在代码之后,我解释了一些细节并回答了奖金问题。

import matplotlib
matplotlib.use('Qt5Agg') #use Qt5 as backend, comment this line for default backend

from matplotlib import pyplot as plt
from matplotlib import animation

fig = plt.figure()

ax = plt.axes(xlim=(0, 2), ylim=(0, 100))

N = 4
lines = [plt.plot([], [])[0] for _ in range(N)] #lines to animate

rectangles = plt.bar([0.5,1,1.5],[50,40,90],width=0.1) #rectangles to animate

patches = lines + list(rectangles) #things to animate

def init():
    #init lines
    for line in lines:
        line.set_data([], [])

    #init rectangles
    for rectangle in rectangles:
        rectangle.set_height(0)

    return patches #return everything that must be updated

def animate(i):
    #animate lines
    for j,line in enumerate(lines):
        line.set_data([0, 2], [10 * j,i])

    #animate rectangles
    for j,rectangle in enumerate(rectangles):
        rectangle.set_height(i/(j+1))

    return patches #return everything that must be updated

anim = animation.FuncAnimation(fig, animate, init_func=init,
                               frames=100, interval=20, blit=True)

plt.show()

说明

我们的想法是绘制您需要的内容,然后重复使用matplotlib返回的艺术家(请参阅更多here)。这是通过首先绘制一个你想要的虚拟草图并保持对象matplotlib给你的。然后在initanimate函数上,您可以更新需要设置动画的对象。

请注意,在plt.plot([], [])[0]我们会收到一位线条艺术家,因此我会使用[plt.plot([], [])[0] for _ in range(N)]收集它们。另一方面,plt.bar([0.5,1,1.5],[50,40,90],width=0.1)返回一个可以为矩形艺术家迭代的容器。 list(rectangles)只需将此容器转换为与lines连接的列表。

我将线条与矩形分开,因为它们的更新方式不同(并且是不同的艺术家),但initanimate会返回所有这些。

回答奖金问题:

  1. line, = plt.plot([], [])plt.plot返回的列表的第一个元素分配给可靠的line
  2. line = plt.plot([], [])只分配整个列表(只有一个元素)。