我有一个的netCDF数据集尺寸[时间,高度,纬度,经度],我已经与xarray打开。我写了一个代码,项目特定时间戳所有的数据到cartopy地图和图像保存到我的目录。我想每一个时间戳创建一个图像,而此刻的唯一途径,我知道如何做到这一点是手动更改的时间戳的条目并再次运行代码。由于有360个时间戳,因此显然需要一些时间。我知道的Python是非常方便的循环,但我很熟悉他们,所以有一个循环中嵌入该代码,这样我可以一气呵成保存多个图像的方式?
pv=data1.pv*10000
pv850=pv[:,0,:,:]
lons=pv850.longitude
lats=pv850.latitude
Fig = plt.figure(figsize=[10,8])
ax = plt.axes(projection=ccrs.NorthPolarStereo())
normi = mpl.Normalize(vmin=-1.5, vmax=12)
cs = ax.contourf(lons, lats, pv850[0,:,:], 50,
transform=ccrs.RotatedPole(), extend='both',
cmap='jet')
plt.colorbar(cs)
ax.coastlines()
ax.gridlines(crs=ccrs.Geodetic(), linestyle='--')
theta = np.linspace(0, 2*np.pi, 100)
center, radius = [0.5, 0.5], 0.5
verts = np.vstack([np.sin(theta), np.cos(theta)]).T
circle = mpath.Path(verts * radius + center)
ax.set_boundary(circle, transform=ax.transAxes)
plt.title('Polar Plot 01-01-2018 00:00')
plt.savefig('image1.png')
行ax.contourf(lons, lats, pv850[0,:,:]
控制时间戳,其中 '0' 对应于时间戳的条目(范围从0-359)。如果可能的话,我还希望标题随每张图的变化而变化。最后,这是最终剧情的照片。
答案 0 :(得分:0)
将其放在一个循环中似乎很简单,所以我不确定为什么这么难。不过,你可以尝试以下。在这里,我将一些定义移到了for循环之外,因为您无需一次又一次地定义360次。
theta = np.linspace(0, 2*np.pi, 100)
center, radius = [0.5, 0.5], 0.5
verts = np.vstack([np.sin(theta), np.cos(theta)]).T
circle = mpath.Path(verts * radius + center)
for i in range(360):
Fig = plt.figure(figsize=[10,8])
ax = plt.axes(projection=ccrs.NorthPolarStereo())
normi = mpl.Normalize(vmin=-1.5, vmax=12)
cs = ax.contourf(lons, lats, pv850[i,:,:], 50,
transform=ccrs.RotatedPole(), extend='both',
cmap='jet')
plt.colorbar(cs)
ax.coastlines()
ax.gridlines(crs=ccrs.Geodetic(), linestyle='--')
ax.set_boundary(circle, transform=ax.transAxes)
plt.title('Polar Plot 01-01-2018 00:{:02d}'.format(i))
plt.savefig('image%s.png' %i)