python - 带 basemap 的 Matplotlib 子图动画

Question

我正在尝试生成温度随时间变化的四面板动画。子图中的四个面板中的每一个都应该是动画 map ；每个面板之间的差异是所使用的数据。我已经设法使用一组数据(没有子图)生成动画，代码如下:

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

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

• 创建您的四个面板:

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

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

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

• 在您的animate 函数中，更改每个轴的颜色。根据数据为 map m_i 分配颜色是使用 m_i.pcolormesh (docs)如您的问题所示。