使用matplotlib动画更新x轴值

时间:2013-07-27 08:08:30

标签: python animation matplotlib

我正在尝试使用matplotlib.ArtistAnimation为两个子图设置动画。我希望随着动画的进行,x轴的值增加,这样动画的总长度就是100,但是在任何时候,子图只向我显示0-24的时间值,然后迭代到100。

给出了一个很好的例子here。该链接使用FuncAnimation并使用plot().axes.set_xlim()以滚动方式更新x轴标签并递增x值。该代码可通过所提供链接中YouTube视频下方的链接获取。

我在下面附加了代码,显示了我尝试复制这些结果,但x-limits似乎采用了最终值而不是随时间递增。我还尝试通过仅绘制将在子图中看到的窗口中的值来递增解决方案(而不是轴),但不会增加x轴值。我也试图实现自动缩放,但x轴仍然没有更新。

我还发现this question几乎是同一个问题,但问题从未得到解答。

这是我的代码:

import matplotlib.pylab as plt
import matplotlib.animation as anim
import numpy as np

#create image with format (time,x,y)
image = np.random.rand(100,10,10)

#setup figure
fig = plt.figure()
ax1=fig.add_subplot(1,2,1)
ax2=fig.add_subplot(1,2,2)
#set up viewing window (in this case the 25 most recent values)
repeat_length = (np.shape(image)[0]+1)/4
ax2.set_xlim([0,repeat_length])
#ax2.autoscale_view()
ax2.set_ylim([np.amin(image[:,5,5]),np.amax(image[:,5,5])])

#set up list of images for animation

ims=[]
for time in xrange(np.shape(image)[0]):

    im = ax1.imshow(image[time,:,:])
    im2, = ax2.plot(image[0:time,5,5],color=(0,0,1))
    if time>repeat_length:
        lim = ax2.set_xlim(time-repeat_length,time)

    ims.append([im, im2])


#run animation
ani = anim.ArtistAnimation(fig,ims, interval=50,blit=False)
plt.show()

我只想要第二个子图(ax2)来更新x轴值。

非常感谢任何帮助。

3 个答案:

答案 0 :(得分:6)

如果需要blitting

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

#create image with format (time,x,y)
image = np.random.rand(100,10,10)

#setup figure
fig = plt.figure()
ax1 = fig.add_subplot(1,2,1)
ax2 = fig.add_subplot(1,2,2)
#set up viewing window (in this case the 25 most recent values)
repeat_length = (np.shape(image)[0]+1)/4
ax2.set_xlim([0,repeat_length])
#ax2.autoscale_view()
ax2.set_ylim([np.amin(image[:,5,5]),np.amax(image[:,5,5])])

#set up list of images for animation


im = ax1.imshow(image[0,:,:])
im2, = ax2.plot([], [], color=(0,0,1))

def func(n):
    im.set_data(image[n,:,:])

    im2.set_xdata(np.arange(n))
    im2.set_ydata(image[0:n, 5, 5])
    if n>repeat_length:
        lim = ax2.set_xlim(n-repeat_length, n)
    else:
        # makes it look ok when the animation loops
        lim = ax2.set_xlim(0, repeat_length)
    return im, im2

ani = animation.FuncAnimation(fig, func, frames=image.shape[0], interval=30, blit=False)

plt.show()

会奏效。

如果你需要跑得更快,你需要使用用于blitting的边界框来玩游戏,以便更新轴标签。

答案 1 :(得分:0)

这会移动您的轴,但速度非常慢。

import matplotlib.pylab as plt
import matplotlib.animation as anim
import numpy as np


image = np.random.rand(100,10,10)
repeat_length = (np.shape(image)[0]+1)/4

fig = plt.figure()
ax1 = ax1=fig.add_subplot(1,2,1)
im = ax1.imshow(image[0,:,:])

ax2 = plt.subplot(122)
ax2.set_xlim([0,repeat_length])
ax2.set_ylim([np.amin(image[:,5,5]),np.amax(image[:,5,5])])
im2, = ax2.plot(image[0:0,5,5],color=(0,0,1))

canvas = ax2.figure.canvas

def init():
    im = ax1.imshow(image[0,:,:])
    im2.set_data([], [])
    return im,im2,

def animate(time):
    time = time%len(image)
    im = ax1.imshow(image[time,:,:])
    im2, = ax2.plot(image[0:time,5,5],color=(0,0,1))
    if time>repeat_length:
        print time
        im2.axes.set_xlim(time-repeat_length,time)
        plt.draw()
    return im,im2,

ax2.get_yaxis().set_animated(True)

# call the animator.  blit=True means only re-draw the parts that have changed.
animate = anim.FuncAnimation(fig, animate, init_func=init,
                               interval=0, blit=True, repeat=True)

plt.show()

答案 2 :(得分:0)

如果您使用blitting,每次更改y / x轴时,都可以调用pyplot.draw()重绘整个图形。

这会更新整个数字,因此相对较慢,但如果你不称它为多项,则可以接受。