matplotlib中的动画标题

时间:2013-07-09 21:00:54

标签: python animation matplotlib

我无法弄清楚如何在FuncAnimation图上使用动画标题(使用blit)。基于http://jakevdp.github.io/blog/2012/08/18/matplotlib-animation-tutorial/Python/Matplotlib - Quickly Updating Text on Axes,我构建了一个动画,但文本部分不会动画。简化示例:

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

vls = np.linspace(0,2*2*np.pi,100)

fig=plt.figure()
img, = plt.plot(np.sin(vls))
ax = plt.axes()
ax.set_xlim([0,2*2*np.pi])
#ttl = ax.set_title('',animated=True)
ttl = ax.text(.5, 1.005, '', transform = ax.transAxes)

def init():
    ttl.set_text('')
    img.set_data([0],[0])
    return img, ttl
def func(n):
    ttl.set_text(str(n))
    img.set_data(vls,np.sin(vls+.02*n*2*np.pi))
    return img, ttl

ani = animation.FuncAnimation(fig,func,init_func=init,frames=50,interval=30,blit=True)

plt.show()

如果删除了blit=True,则会显示文字,但速度会慢下来。它似乎与plt.titleax.set_titleax.text失败。

编辑:我发现为什么第一个链接中的第二个例子有效;文本在img部分内。如果您将上述1.005设为.99,您就会明白我的意思。可能有一种方法可以用边界框来做这件事,不知怎的......

4 个答案:

答案 0 :(得分:16)

请参阅Animating matplotlib axes/tickspython matplotlib blit to axes or sides of the figure?

所以,问题在于animation胆量背景实际保存的内容(animation.py的第792行),它抓住了轴中的内容边界框。当您有多个轴独立动画时,这是有意义的。在您的情况下,您只需要担心一个axes,我们希望为轴边框的设置动画。通过一些猴子修补,一定程度的容忍度来达到mpl的内脏并且稍微捅一下,接受最快和最肮脏的解决方案我们可以解决你的问题:

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

def _blit_draw(self, artists, bg_cache):
    # Handles blitted drawing, which renders only the artists given instead
    # of the entire figure.
    updated_ax = []
    for a in artists:
        # If we haven't cached the background for this axes object, do
        # so now. This might not always be reliable, but it's an attempt
        # to automate the process.
        if a.axes not in bg_cache:
            # bg_cache[a.axes] = a.figure.canvas.copy_from_bbox(a.axes.bbox)
            # change here
            bg_cache[a.axes] = a.figure.canvas.copy_from_bbox(a.axes.figure.bbox)
        a.axes.draw_artist(a)
        updated_ax.append(a.axes)

    # After rendering all the needed artists, blit each axes individually.
    for ax in set(updated_ax):
        # and here
        # ax.figure.canvas.blit(ax.bbox)
        ax.figure.canvas.blit(ax.figure.bbox)

# MONKEY PATCH!!
matplotlib.animation.Animation._blit_draw = _blit_draw

vls = np.linspace(0,2*2*np.pi,100)

fig=plt.figure()
img, = plt.plot(np.sin(vls))
ax = plt.axes()
ax.set_xlim([0,2*2*np.pi])
#ttl = ax.set_title('',animated=True)
ttl = ax.text(.5, 1.05, '', transform = ax.transAxes, va='center')

def init():
    ttl.set_text('')
    img.set_data([0],[0])
    return img, ttl

def func(n):
    ttl.set_text(str(n))
    img.set_data(vls,np.sin(vls+.02*n*2*np.pi))
    return img, ttl

ani = animation.FuncAnimation(fig,func,init_func=init,frames=50,interval=30,blit=True)

plt.show()

请注意,如果图中有多个轴,则可能无法按预期工作。一个更好的解决方案是将axes.bbox 扩展到足以捕获标题+轴刻度标签。我怀疑在mpl中有代码可以做到这一点,但我不知道它在哪里。

答案 1 :(得分:5)

要添加到tcaswell的“猴子修补”解决方案,以下是如何将动画添加到轴刻度标签。具体来说,要为x轴设置动画,请设置FileInputStream并从动画函数返回ax.xaxis.set_animated(True)

ax.xaxis

答案 2 :(得分:1)

你必须致电

plt.draw()

ttl.set_text(str(n))

这里有一个非常简单的文字动画示例,在“没有FuncAnimation()”的图形中。试试吧,你会发现它对你有用。

import matplotlib.pyplot as plt
import numpy as np
titles = np.arange(100)
plt.ion()
fig = plt.figure()
for text in titles:
    plt.clf()
    fig.text(0.5,0.5,str(text))
    plt.draw()

答案 3 :(得分:1)

如果您需要修改标题,可以使用以下命令更新标题:

fig.suptitle()

见图API document