[PYTHON]在一個figure中畫兩組圖,以及兩個colorbar

畫出來效果是這樣的


好美對不對?

左邊和右邊的圖對應的colorbar是不同的。

如果每一個子圖也畫一個colorbar那會很浪費空間,所以在最底下的地方畫colorbar,因此要自己手動設colorbar的位置,主要是用這個參數cax=plt.axes([0.65, 0.06, 0.3, 0.02])

不多說了,上代碼了。

def manycs2():
  N1=7
  N2=2
  ww = 0.4
  hh = 0.12
  for i_hour in range(0,25,1):
    fig=plt.figure()
    for i1 in range(0,N1):
      for i2 in range(0,N2):
        print(i_hour,i1,i2)
        da=nc.Dataset("C:\\Users\\kong\\PycharmProjects\\southcnseapycode\\crosssection\\4draw_crosssection_%d"%(i1+1))
        u=da.variables["u"][:]
        v=da.variables["v"][:]
        z=da.variables["z"][:]
        w=da.variables["w"][:]
        llj=da.variables["llj"][:]
        lat=da.variables["lat"][:]
        lon=da.variables["lon"][:]
        lon001=da.variables["lon001"][:]
        lat001=da.variables["lat001"][:]
        theta=da.variables["theta"][:]
        da.close()
        uv1=u*np.cos(theta)+v*np.sin(theta)##x
        uv2=-u*np.sin(theta)+v*np.cos(theta)##y
        lonlon=np.zeros((np.shape(z)[1],np.shape(lon)[0]))
        for i in range(0,np.shape(z)[1]):
          lonlon[i,:]=lon

        #--u--#
        a = fig.add_axes([0.075+i2*ww, 0.10 + i1*hh, ww, hh])
        if i2==0:
          cslev = np.arange(-6,7,1)#13
          cs=plt.contourf(lonlon,z[i_hour,:,:],uv1[i_hour,:,:],cslev,cmap=cm.coolwarm,extend='both')
        else:
          cslev = np.arange(0,10,1)#13
          cs=plt.contourf(lonlon,z[i_hour,:,:],uv2[i_hour,:,:],cslev,cmap=cm.coolwarm,extend='both')
        cs.cmap.set_over('k')
        cs.cmap.set_under('b')
        #cbar = plt.colorbar(cs)
        #cbar.set_label('u\'(m/s)')
        plt.ylim([0,4000])
        if i2==0:
          plt.yticks([0,1000,2000,3000],size=10)
        else:
          plt.yticks([],size=10)
        plt.xticks(size=5)
        ##--vector--##
        vectorsize=5
        rho=3
        Q = plt.quiver(lonlon[:,::rho],z[i_hour,:,::rho],uv1[i_hour,:,::rho],100000*w[i_hour,:,::rho],width=vectorsize,headwidth=vectorsize*2/3,headlength=vectorsize/2,headaxislength=vectorsize/2,minlength=0,units='dots',angles='xy', scale_units='xy', scale=100,minshaft=0.5)#100
        if i1==6:
          if i2==0:
            cbar=fig.colorbar(cs,orientation='horizontal',cax=plt.axes([0.05, 0.06, 0.3, 0.02]))
            fig.text(0.35, 0.03,'u\'(m/s)',size=15)
          else:
            cbar=fig.colorbar(cs,orientation='horizontal',cax=plt.axes([0.65, 0.06, 0.3, 0.02]))
            fig.text(0.55, 0.03,'v\'(m/s)',size=15)
    if i_hour==24:
      fig.text(0.4, 0.95,"average",size=20)
    elif i_hour+7<24:
      fig.text(0.4, 0.95,"%d UTC, %d localtime"%(i_hour,i_hour+7),size=20)
    else:
      fig.text(0.4, 0.95,"%d UTC, %d localtime"%(i_hour,i_hour+7-24),size=20)
    fig.text(0.45, 0.03,"lon",size=30)
    plt.savefig('C:\\Users\\kong\\PycharmProjects\\southcnseapycode\\crosssection\\crosssection%d hour'%i_hour,dpi=500)
    plt.close()


  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 在Python中,我们可以使用Matplotlib库来创建子图并共用一个colorbar。下面是一个示例代码: ```python import matplotlib.pyplot as plt import numpy as np # 创建一个2x2的子图布局 fig, axes = plt.subplots(2, 2, figsize=(8, 6)) # 生成一些示例数据 data1 = np.random.rand(10, 10) data2 = np.random.rand(10, 10) data3 = np.random.rand(10, 10) # 在第一个子图中绘制热图 im1 = axes[0, 0].imshow(data1, cmap='viridis') axes[0, 0].set_title('Subplot 1') # 在第二个子图中绘制热图 im2 = axes[0, 1].imshow(data2, cmap='viridis') axes[0, 1].set_title('Subplot 2') # 在第三个子图中绘制热图 im3 = axes[1, 0].imshow(data3, cmap='viridis') axes[1, 0].set_title('Subplot 3') # 创建一个共用的colorbar cbar = fig.colorbar(im3, ax=axes.ravel().tolist(), shrink=0.6, aspect=20) cbar.set_label('Colorbar') # 调整子图之间的间距 fig.tight_layout() # 显示图形 plt.show() ``` 在上述代码中,我们通过`plt.subplots`创建了一个2x2的子图布局,并使用`fig.colorbar`函数为其中的第一个子图创建了一个共用的colorbar。`ax=axes.ravel().tolist()`参数指定了要在哪些子图中显示colorbar,`shrink`和`aspect`参数用于调整colorbar的大小和纵横比。最后,使用`tight_layout`函数调整子图之间的间距,并通过`plt.show`显示图形。 这样,我们就实现了三个子图共用一个colorbar的功能。注意,上述示例中的示例数据都是随机生成的,你可以根据自己的数据进行相应的修改和适应。 ### 回答2: 在Python中,我们可以使用matplotlib库来绘制图形和颜色条。要实现三个子图共用一个颜色条,可以按照以下步骤操作: 首先,导入所需的库: ```python import matplotlib.pyplot as plt import numpy as np ``` 然后,创建三个子图并生成数据: ```python fig, axs = plt.subplots(1, 3, figsize=(12, 4)) # 创建一个包含三个子图的画布 data1 = np.random.rand(10, 10) # 生成第一个子图的数据 data2 = np.random.rand(10, 10) # 生成第二个子图的数据 data3 = np.random.rand(10, 10) # 生成第三个子图的数据 ``` 接下来,将三个子图绘制出来: ```python im1 = axs[0].imshow(data1, cmap='hot') # 绘制第一个子图,并将颜色条保存在im1中 im2 = axs[1].imshow(data2, cmap='hot') # 绘制第二个子图,并将颜色条保存在im2中 im3 = axs[2].imshow(data3, cmap='hot') # 绘制第三个子图,并将颜色条保存在im3中 ``` 然后,确定颜色条的位置和大小,并将其添加到画布上: ```python cbar_ax = fig.add_axes([0.92, 0.3, 0.02, 0.4]) # 确定颜色条的位置和大小 fig.colorbar(im1, cax=cbar_ax) # 将颜色条添加到画布上,并指定参考图像为im1 ``` 最后,显示图形: ```python plt.show() ``` 这样,就实现了三个子图共用一个颜色条的效果。 ### 回答3: 三个子图能够共用一个colorbar,可以通过matplotlib中的gridspec来实现。gridspec提供了更灵活的子图布局方式。 首先,我们创建一个包含三个子图的网格(spec)。我们可以通过gridspec的`subplots`方法来实现: ```python import matplotlib.pyplot as plt import matplotlib.gridspec as gridspec # 创建一个包含三个子图的网格,每行一个子图 gs = gridspec.GridSpec(3, 1) # 创建第一个子图 ax1 = plt.subplot(gs[0]) # 创建第二个子图 ax2 = plt.subplot(gs[1]) # 创建第三个子图 ax3 = plt.subplot(gs[2]) ``` 接下来,我们可以在每个子图上绘制不同的数据,并为每个子图设置相同的colorbar。这可以通过调用`colorbar`方法并传入子图对象以及关联的绘图对象(例如图像、散点图等)来实现。 ```python # 在第一个子图上绘制数据1 ax1.imshow(data1) # 在第二个子图上绘制数据2 ax2.imshow(data2) # 在第三个子图上绘制数据3 ax3.imshow(data3) # 创建共享colorbar cbar = plt.colorbar(ax=axs, orientation="vertical") ``` 这样就可以在三个子图之间共享一个colorbar了。需要注意的是,共享colorbar应该基于相同范围的数据创建(例如,数据1、数据2和数据3的值范围应相同),以保证颜色映射的一致性。 最后,通过调用`plt.show()`展示出这三个子图以及共享colorbar。 ```python plt.show() ``` 这样,我们就成功地在三个子图之间实现了共享一个colorbar的效果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值