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），但仍然不知道如何使用 Basemap 进行操作。

Answer 1

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

• 创建您的四个面板：

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

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

• 当您创建四个 Basemap 实例时，使用 ax参数(docs)。

• 在您的 animate 函数中，更改每个轴的颜色。根据数据分配颜色以映射 m_i 与 m_i.pcolormesh (docs) 如您的问题所示。