我正在尝试执行以下操作:绘制点并将引用存储在字典中。动画删除点动画。一个最小的例子如下:
%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),这是有效的。但如果两者都被绘制,则不会删除第一个点,也不会删除第二个点。
相关问题让我感觉就像现在一样:
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。
答案 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()