带底图的 Matplotlib 子图动画

Question

我正在尝试生成温度随时间变化的四面板动画。子图中的四个面板中的每一个都应该是一个动画地图；每个面板之间的差异在于所使用的数据。我已成功使用一组数据（没有子图）和以下代码生成动画：

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

#dummy temperature data with 10 time-steps
y=np.random.randn(10, 60, 100) 

fig = plt.figure()
m = Basemap(projection='kav7',lon_0=0)
lats=np.linspace(90,-90,y.shape[1])
lons=np.linspace(-180,180,y.shape[2])
lons, lats = np.meshgrid(lons,lats)

m.drawparallels(np.arange(-90.,99.,30.), labels=[1,0,0,0])
m.drawmeridians(np.arange(-180.,180.,60.), labels=[0,0,0,1])
m.drawcoastlines(linewidth=0.25)
m.pcolormesh(lons,lats,y[0],cmap=plt.cm.bwr, shading='flat',latlon=True)

def init():
    m                                                                                                                                                                                                                                                                                                 

def animate(i):
    m.pcolormesh(lons,lats,y[i],cmap=plt.cm.bwr, shading='flat',latlon=True)
    return m

anim = animation.FuncAnimation(fig, animate, init_func=init, frames=10, interval=100) #interval = number of milliseconds between frames                                                                                                                                                                                                                                                                                                                       

anim.save('movie.mp4')

我已经看过许多子图动画的示例（1、2、3 ） ，但仍然不知道如何使用底图来实现它。

Answer 1

您的动画函数需要更新图中四个不同轴上的四个不同贴图。

• 创建四个面板：

  fig, ax = plt.subplots(2, 2)
  

轴被索引。我建议将它们放置在 2x2 布局中，它们被索引为二维数组，因此在创建地图时将轴引用为ax[0][0]、ax[0][1]、ax[1][0]、 和。ax[1][1]

• 创建底图的四个实例时，使用 ax参数(docs)将每个地图分配给不同的轴。

• 在您的animate函数中，更改每个轴的颜色。如您的问题所示，根据数据分配颜色m_i以映射到m_i.pcolormesh （文档） 。