15

我正在尝试为一些时间相关数据的空间坐标的 wigner 函数设置动画。wigner 函数是二维的，所以我使用 contourf() 来绘制它。我将数据存储在 HDF5 文件中，并且可以即时制作 Wigner 分布，但我不知道如何对其进行动画处理。我能找到的所有动画教程和示例（例如这个和这个）都严格用于线图。具体来说，他们的animate(i)功能使用line.set_data()，我似乎无法找到等效的contourf().

如何为使用 制作的图像制作动画contourf()？

contourf()相当于什么set_data()？

4

4 回答 4

13

有一个简单的方法来做到这一点FuncAnimation：你必须有一个函数来清除轴并根据帧号绘制一个新的轮廓。不要忘记设置blit为False.

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

DATA = np.random.randn(800).reshape(10,10,8)


fig,ax = plt.subplots()

def animate(i):
       ax.clear()
       ax.contourf(DATA[:,:,i])
       ax.set_title('%03d'%(i)) 

interval = 2#in seconds     
ani = animation.FuncAnimation(fig,animate,5,interval=interval*1e+3,blit=False)

plt.show()
于 2016-07-15T17:10:26.377 回答
7

我正在绘制地理数据，因此需要底图。根据Captain_M的回答和https://github.com/matplotlib/matplotlib/issues/6139上的讨论/错误报告， 我发布了一个受tacaswell启发的响应，允许您在二维数据动画中使用 contourf 并保存如果你有 ffmpeg，它作为 mp4：

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


fig, ax = plt.subplots()

# set up map projection
m = Basemap(projection='nsper',lon_0=-0,lat_0=90)
m.drawcoastlines()
m.drawparallels(np.arange(0.,180.,30.))
m.drawmeridians(np.arange(0.,360.,60.))

# some 2D geo arrays to plot (time,lat,lon)
data = np.random.random_sample((20,90,360))
lat = np.arange(len(data[0,:,0]))
lon = np.arange(len(data[0,0,:]))
lons,lats = np.meshgrid(lon,lat)

# ims is a list of lists, each row is a list of artists to draw in the
# current frame; here we are animating three artists, the contour and 2 
# annotatons (title), in each frame
ims = []
for i in range(len(data[:,0,0])):
    im = m.contourf(lons,lats,data[i,:,:],latlon=True)
    add_arts = im.collections
    text = 'title={0!r}'.format(i)
    te = ax.text(90, 90, text)
    an = ax.annotate(text, xy=(0.45, 1.05), xycoords='axes fraction')
    ims.append(add_arts + [te,an])

ani = animation.ArtistAnimation(fig, ims)
## If you have ffmpeg you can save the animation by uncommenting 
## the following 2 lines
# FFwriter = animation.FFMpegWriter()
# ani.save('basic_animation.mp4', writer = FFwriter)
plt.show()
于 2016-03-10T16:42:41.270 回答
6

这是我用来制作二维等高线图的动画，它改编自http://matplotlib.org/examples/animation/dynamic_image2.html

import pylab as pl
import numpy as np
import matplotlib.animation as animation
import types


fig = pl.figure()
# Some 2D arrays to plot (time,x,y)
data = np.random.random_sample((20,10,10))

# ims is a list of lists, each row is a list of artists to draw in the
# current frame; here we are just animating one artist, the image, in
# each frame
ims = []
for i in range(len(data[:,0,0])):
    t_step = int(i)
    im = pl.contourf(data[i,:,:])

    #################################################################
    ## Bug fix for Quad Contour set not having attribute 'set_visible'
    def setvisible(self,vis):
        for c in self.collections: c.set_visible(vis)
    im.set_visible = types.MethodType(setvisible,im)
    im.axes = pl.gca()
    im.figure=fig
    ####################################################################

    ims.append([im])

ani = animation.ArtistAnimation(fig, ims, interval=70, blit=False,repeat_delay=1000)

pl.show()
于 2014-05-01T01:35:43.090 回答
1

如果您像我一样并且 matplotlib.animation 不起作用。这是您可以尝试的其他方法。如果要不断更新颜色条和图中的其他所有内容，请在一开始使用 plt.ion() 以启用交互式绘图，并使用 plt.draw() 和 plt.clf() 的组合来不断更新绘图. 这是示例代码：

import matplotlib.pyplot as plt
import numpy as np

plt.ion(); plt.figure(1);
for k in range(10):
    plt.clf(); plt.subplot(121);
    plt.contourf(np.random.randn(10,10)); plt.colorbar();
    plt.subplot(122,polar=True)
    plt.contourf(np.random.randn(10,10)); plt.colorbar();
    plt.draw();

请注意，这适用于包含不同子图和各种类型图（即极坐标或笛卡尔坐标）的图

于 2015-07-05T04:39:00.330 回答