Python气象数据处理与绘图(9):更自由的多子图组图绘制

今天分享一个比较实用的多子图绘制方法,由于气象科研中经常需要将多图拼成组图,并且需要将多种类型的图拼在一起(比如EOF,将填色图和折线图拼在一起),虽说在前几篇文章均有子图的设置代码,但是并未做过多讲解,我所用的方法与matplotlib官方库提供的例子也有些不同,今天便详细分析一下我所使用的方法和官方提供的方法的差异。
随便找到了一张我之前画过的组图


example

可以看到,该组图的组合较为随意,是两张填色图,两张折线图的组合,并且对两张填色图设置了统一色标。然而这样的子图组合方式,在matplotlib的例子中并未直接给出。
官方关于组图提供了多种方法,然而常见的方法如下:
首先给出官方的第一种方法:

ax1 = plt.subplot(212)
ax2 = plt.subplot(221)
ax3 = plt.subplot(222)
plt.show()
ex.1
plt.subplot(xyz)

通过该语句控制子图的数量和位置,x为子图行数,y为子图列数,z为索引号,从左到右从上到下,如果用来绘制同样属性,同样大小的多个子图的组图,用此方法十分方便,比如:
https://matplotlib.org/gallery/lines_bars_and_markers/scatter_star_poly.html#sphx-glr-gallery-lines-bars-and-markers-scatter-star-poly-py
然而,比如说一堆一样的子图中出了一个叛徒

import numpy as np
import matplotlib.pyplot as plt
np.random.seed(19680801)
x = np.random.rand(10)
y = np.random.rand(10)
z = np.sqrt(x**2 + y**2)
plt.subplot(321)
plt.scatter(x, y, s=80, c=z, marker=">")
plt.subplot(322)
plt.scatter(x, y, s=80, c=z, marker=(5, 0))
verts = np.array([[-1, -1], [1, -1], [1, 1], [-1, -1]])
plt.subplot(323)
plt.scatter(x, y, s=80, c=z, marker=verts)
plt.subplot(324)
plt.scatter(x, y, s=80, c=z, marker=(5, 1))
plt.subplot(325)
plt.scatter(x, y, s=80, c=z, marker='+')
plt.subplot(326)
plt.scatter(x, y, s=80, c=z, marker=(5, 2))
plt.colorbar()
plt.show()

给每一个子图都添加一个色标显得很累赘,通常文献中此类图共享一个色标,因此,当我们给最后一个子图添加色标时,便出现了这样的问题:


ex2

不管是不是强迫症都难受死了,或许有人说,干嘛单独给一个子图添加色标,可以给整个画布添加色标呀,但是比如说你时将两类图拼在一起,那么给每类图的子图添加色标就十分必要了,因此在这种情况下,这个方法是不适用的。同样,官方的另一个例子用这种方法也是不合适的:

gridspec.GridSpec()

详见:https://matplotlib.org/gallery/subplots_axes_and_figures/align_labels_demo.html#sphx-glr-gallery-subplots-axes-and-figures-align-labels-demo-py
以上两种方法我再最初使用后难以忍受,曾经一度想python处理数据,ncl绘图,后来偶然的发现了一种方法,可以十分自由的定义子图。

fig = plt.figure(figsize=(12,7))   

首先创建画图,接着使用

a = fig.add_axes([x, y, i, j])

来创建第一个子图,其中x,y是该子图在画布中的坐标,i,j是该子图的宽和高。坐标相对于画布,可以理解为直角坐标系的第一象限,用图直观的看就是:



那多个子图则是:

a = fig.add_axes([x1, y1, i1, j1])
b = fig.add_axes([x2, y2, i2, j2])
c = fig.add_axes([x3, y3, i3, j3])

以此类推,麻烦的点在于需要不断调整坐标位置及子图的宽高,但是这也正是该方法的自由之处,你可以任意规划你的组图配置,这种方法在jupyter notebook上操作更是方便至极。
现在我给出本文一开始展示的图片的部分代码,数据部分略过了,只展示关键的画图部分。

#设置填色图投影方式,以及地图边界
proj = ccrs.PlateCarree(central_longitude=80)
leftlon, rightlon, lowerlat, upperlat = (10,140,20,70)
img_extent = [leftlon, rightlon, lowerlat, upperlat]
#建立画布
fig2 = plt.figure(figsize=(15,15))
#添加第一个子图,左上角图,注意c1是填色图图层
f2_ax1 = fig2.add_axes([0.1, 0.24, 0.5, 0.5],projection = proj)
contour_map(f2_ax1,img_extent,10)
f2_ax1.set_title('(a)',loc='left')
f2_ax1.set_title( '%.2f%%' % (var[0]*100),loc='right')
f2_ax1.add_feature(cfeature.OCEAN,alpha=1,facecolor='w', zorder=1)
f2_ax1.add_feature(cfeature.COASTLINE.with_scale('50m'), zorder=2) 
c1 = f2_ax1.contourf(lon_region,lat_region, eof[0,:,:], levels=np.arange(-0.9,1.0,0.1), zorder=0, extend = 'both', transform=ccrs.PlateCarree(), cmap=plt.cm.RdBu_r)
#添加第二个子图,右上角图,注意c3是填色图图层,要使用统一色标,需C3和C1的levels相同
f2_ax2 = fig2.add_axes([0.65, 0.24, 0.5, 0.5],projection = proj)
contour_map(f2_ax2,img_extent,10)
f2_ax2.set_title('(b)',loc='left')
f2_ax2.set_title( '%.2f%%' % (var[1]*100),loc='right')
f2_ax2.add_feature(cfeature.OCEAN,alpha=1,facecolor='w', zorder=1)
f2_ax2.add_feature(cfeature.COASTLINE.with_scale('50m'), zorder=2) 
c3 = f2_ax2.contourf(lon_region,lat_region, eof[1,:,:], levels=np.arange(-0.9,1.0,0.1), zorder=0, extend = 'both', transform=ccrs.PlateCarree(), cmap=plt.cm.RdBu_r)
#添加第三个子图,左下角图
f2_ax3 = fig2.add_axes([0.1, 0.1, 0.5, 0.2])
f2_ax3.set_title('(c)',loc='left')
plt.ylim(-4,4)
f2_ax3.axhline(0,linestyle="--")
f2_ax3.bar(years,pc[:,0],color=color1)
f2_ax3.plot(years,pc2[:,0],color="b")
#添加第四个子图,左下角图
f2_ax4 = fig2.add_axes([0.65, 0.1, 0.5, 0.2])
f2_ax4.set_title('(d)',loc='left')
plt.ylim(-4,4)
f2_ax4.axhline(0,linestyle="--")
f2_ax4.bar(years,pc[:,1],color=color2)
f2_ax4.plot(years,pc2[:,1],color="b")
#添加色标,position定义色标位置,c1指定从c1填色图层取色,由于C3,C1的levles相同,所以色标一致,orientation设置色标为水平还是垂直,format设置色标标签格式
position=fig2.add_axes([0.38, 0.33, 0.5, 0.017])
fig2.colorbar(c1,cax=position,orientation='horizontal',format='%.1f',)
plt.show()

再次展示绘图效果方便对比:


可以看到该方法可以定义任意部分的位置,使得子图的拼接更加自由,同样需要注意的是,填色图的宽度高度是0.5,0.5,而折线图的高度是0.5,0.2,然而画出来的效果却是差不多的,这是由于填色图叠加了地图底图,坐标经过了投影换算,因此存在差异,因此具体的每个子图的大小和位置是需要不断调整得出的。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容