删除matplotlib底图动画中存储在字典中的点

时间:2016-02-19 09:45:53

标签: python animation matplotlib plot matplotlib-basemap

我正在尝试执行以下操作:绘制点并将引用存储在字典中。动画删除点动画。一个最小的例子如下:

%matplotlib qt
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
import numpy as np
import matplotlib.animation as animation

fig = plt.figure()  
m = Basemap(projection='aeqd',lat_0=72,lon_0=29, resolution='l',   
        llcrnrlon=15, llcrnrlat=69,
        urcrnrlon=41, urcrnrlat=75.6,area_thresh = 100)

pointDict=dict()

pointDict[1]=m.plot (0, 0,marker='.',label='first')[0]
pointDict[2]=m.plot (0, 0,marker='.',label='second')[0]

def init():
    print ("Init")
    x,y = m(30, 73)
    pointDict[1].set_data(x,y)
    x,y = m(31, 73)
    pointDict[2].set_data(x,y)
    return pointDict.values()

def animate(i):
    print ("Frame {0}".format(i))
    if i==2:
        l=pointDict.pop(1)
        print ("Removing {0}".format(l.get_label()))
        l.remove()
        del l
    return pointDict.values()

anim = animation.FuncAnimation(plt.gcf(), animate, init_func=init,
                           frames=10, interval=1000, blit=True)

plt.show()

输出:

Init
Init
Frame 0
Frame 1
Frame 2
Removing first
Frame 3

有趣的是,如果我只绘制第一个点(即在init函数中删除pointDict [2] = m.plot和pointDict [2] .set_data),这是有效的。但如果两者都被绘制,则不会删除第一个点,也不会删除第二个点。

相关问题让我感觉就像现在一样:

Matplotlib Basemap animation

How to remove lines in a Matplotlib plot

Matplotlib animating multiple lines and text

Python, Matplotlib, plot multi-lines (array) and animation

我正在使用带有Python-2.7内核的Anaconda。

1 个答案:

答案 0 :(得分:0)

我发现了问题所在,因此希望自己回答我的问题: 这个问题有点出乎意料blit = True。 显然,只有在动画函数中设置了点时才能使用blitting。因此,在init例程中设置数据会导致问题。 所以有两个选择:将blit设置为False,但这不是很优雅。另一种选择是在第一帧中设置点。 然后使用的init和animate函数如下:

def init():
    print ("Init")
    pointDict[1].set_data([],[])
    pointDict[2].set_data([],[])
    return pointDict.values()

def animate(i):
    print ("Frame {0}".format(i))
    if i==0:
        print ("Init")
        x,y = m(30, 73)
        pointDict[1].set_data(x,y)
        x,y = m(31, 73)
        pointDict[2].set_data(x,y)

    if i==2:
        l=pointDict.pop(1)
        print ("Removing {0}".format(l.get_label()))
        l.remove()
        del l
    return pointDict.values()

anim = animation.FuncAnimation(plt.gcf(), animate, init_func=init,
                       frames=10, interval=1000, blit=True)

plt.show()