Matplotlib Subplot动画与底图

时间:2015-11-01 01:56:27

标签: python animation matplotlib matplotlib-basemap subplot

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

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')

我查看了很多子动画动画的示例(123),但仍然不知道如何使用Basemap。

1 个答案:

答案 0 :(得分:1)

您的动画功能需要在图中的四个不同轴上更新四个不同的地图。

  • 创建四个面板:

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

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

  • 创建四个Basemap实例时,使用。将每个地图指定给不同的轴 ax参数(docs)

  • animate功能中,更改每个轴的颜色。如您的问题所示,根据要映射m_i的数据分配颜色与m_i.pcolormesh (docs)一致。