Pyplot zorder失去动画

时间:2014-09-15 10:03:33

标签: python animation matplotlib z-order

我试图使用pyplot的funcanimation来动画同一图表中的不同对象 除了显示不同元素的顺序之外,它几乎按照我的预期工作。因此,绘图曲线,文本和图例显示在图像后面,几乎看不到它们。

这是我的(不是那么)最小的工作示例:

#! /usr/bin/python
import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation
import random

def init():
    line_simu.set_data([], [])
    time_text.set_text('')
    imobj.set_data(np.zeros((100, 100)))
    imobj.set_zorder(0)
    time_text.set_zorder(10)
    return line_simu,  time_text, imobj

def animate(i):
    imobj.set_zorder(0)
    time_text.set_zorder(10)
    y_simu = np.linspace(0,100, 100)
    x_simu =    np.linspace(-10, 10, 100) 
    line_simu.set_data(x_simu, y_simu)

    time_text.set_text('time = %.1f' % i )

    global data
    imobj.set_data( data + np.random.random((100,1)) * 0.5 )

    return line_simu, time_text, imobj


def forceAspect(ax,aspect=1):
    im = ax.get_images()
    extent =  im[0].get_extent()
    ax.set_aspect(abs((extent[1]-extent[0])/(extent[3]-extent[2]))/aspect)


fig = plt.figure()
ax = plt.axes(xlim=(-15,15), ylim=(-110, 0) , aspect=1)

data = np.random.random((100,100)) - .5
imobj = ax.imshow( data , extent=[-15,15, -110, 0.0], origin='lower', cmap=plt.cm.gray, vmin=-2, vmax=2, alpha=.7, zorder=0, aspect=1)

line_simu, = ax.plot([], [],"r--", lw=2, markersize=4 , label = "Some curve" ,  zorder= 2 )

time_text = ax.text(-14.9, -108, '', zorder=10)

l = plt.legend(loc='lower right', prop={'size':8} )
l.set_zorder(200)

forceAspect(ax,aspect=1)

anim = animation.FuncAnimation(fig, animate, init_func=init,  frames=range( 50), interval=3000, blit=True)

plt.show()

没有动画,我可以使用set_zorder轻松控制不同元素的顺序,但是当动画更新图像时,此顺序将丢失。我试图在init函数中设置zorder并再次在animate函数中设置,但没有成功。

我非常感谢你对此事的任何帮助。

1 个答案:

答案 0 :(得分:1)

回答我自己的问题:似乎init()和animate()函数返回对象的顺序控制它们的显示顺序。此外,这些函数应返回图例对象,以便将其包含在动画中。

这是我更正后的代码:

#! /usr/bin/python
import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation
import random

def init():
    imobj.set_data(np.zeros((100, 100)))
    line_simu.set_data([], [])
    time_text.set_text('time = 0.0')

    return imobj , line_simu,  time_text,  l

def animate(i):
    global data
    imobj.set_data( data + np.random.random((100,1)) * 0.5 )

    imobj.set_zorder(0)
    y_simu = np.linspace(-100,-10, 100)
    x_simu = np.linspace(-10, 10, 100) 
    line_simu.set_data(x_simu, y_simu)
    time_text.set_text('time = %.1f' % i )


    return imobj , line_simu, time_text,  l


def forceAspect(ax,aspect=1):
    im = ax.get_images()
    extent =  im[0].get_extent()
    ax.set_aspect(abs((extent[1]-extent[0])/(extent[3]-extent[2]))/aspect)



fig = plt.figure()
ax = plt.axes(xlim=(-15,15), ylim=(-110, 0) , aspect=1)

data = np.random.random((100,100)) - .5
imobj = ax.imshow( data , extent=[-15,15, -110, 0.0], origin='lower', cmap=plt.cm.gray, vmin=-2, vmax=2, alpha=1.0, zorder=1, aspect=1)

line_simu, = ax.plot([], [],"r--", lw=2, markersize=4 , label = "Some curve" ,  zorder= 1 )
time_text = ax.text(-14.0, -108, '', zorder=10)

forceAspect(ax,aspect=1)

l = plt.legend(loc='lower right', prop={'size':8} )

anim = animation.FuncAnimation(fig, animate, init_func=init,  frames=range( 50), interval=500, blit=True)

plt.show()