多个子图右侧的唯一颜色条

时间:2020-03-25 12:10:25

标签: matplotlib subplot colorbar

我正在尝试向一组2x2子图添加唯一的颜色条,我希望它位于集合的右侧。正如您在代码的最后几行中看到的那样,我已经能够生成颜色条,但是结果是它显示在最后一个面板的右侧(请参见下图)。有人可以帮我找到解决方案吗?

import os
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
from matplotlib import cm
import pandas as pd
mark=['-o','-s','-o','-^','-s']
Axs=['ax1','ax2','ax3','ax4']
RadiLabel=['5','7','10','15']
Lambda=np.arange(10,20,1)
print(Lambda[:])

Num_Colors=int(len(Lambda))
print(Num_Colors)
cm = plt.get_cmap('viridis_r')
# viridis,plasma,inferno,magma,gist_rainbow,gnuplot,autumn,Blues

mpl.rcParams['xtick.direction'] = 'in'
mpl.rcParams['ytick.direction'] = 'in'
mpl.rcParams['xtick.major.size'] = 15
mpl.rcParams['xtick.minor.size'] = 5
mpl.rcParams['ytick.major.size'] = 15
mpl.rcParams['ytick.minor.size'] = 5
mpl.rcParams['font.family'] = 'Calibri'

fig = plt.figure(figsize=(12,9))
#fig.subplots_adjust(right=0.8)

ax=fig.add_subplot(111)
ax.set_title('Area', fontsize=20)
plt.axis([0.8,15.2, 0.,0.3])

File= 'SCAli.txt'
data=np.loadtxt(File)
dat=pd.read_fwf("SCAli.txt",header=None,names=["Area","R","L","H"])

k=0
#print(len(Axs))
for j in range(0,len(Axs)):
    k=k+1
    axes=Axs[j]=plt.subplot(2,2,k)
    Axs[j].set_prop_cycle(color=[cm(1.*ii/Num_Colors) for ii in range(Num_Colors)])
    Axs[j].annotate('R= '+RadiLabel[j]+' nm', xy=(0.07, 0.9), xycoords="axes fraction",fontsize=20)

    for i in Lambda:
        dat2=dat.loc[(dat['L'] == i) & (dat['R'] == Radi[j])]
        Axs[j].plot(dat2['H'],dat2['Area'],mark[0],markersize=10,markeredgecolor='black',lw=2)
        plt.xlabel("Height", fontsize=30)
        plt.ylabel("Area", fontsize=30)
        plt.xticks(fontsize=20, rotation=0)
        plt.yticks(fontsize=20, rotation=0)
        Axs[j].minorticks_off()

plt.tight_layout()

# Here we plot the colorbar
sm = plt.cm.ScalarMappable(cmap=cm, norm=plt.Normalize(vmin=1, vmax=2))
cbar=plt.colorbar(sm)
cbar.ax.tick_params(labelsize=20)
cbar.set_label('$\lambda$',size=20)

plt.show()
plt.close()

This is the figure I get

Link to the file SCAli.txt

1 个答案:

答案 0 :(得分:0)

我找到了我问题的答案。经过一些试验并查看了类似的示例,我意识到添加以下代码行就足够了:

    plt.subplots_adjust(bottom=0., right=0.92, top=1.)
    cax = plt.axes([0.95, 0.3, 0.025, 0.4])
    sm = plt.cm.ScalarMappable(cmap=cm, norm=plt.Normalize(vmin=1, vmax=2))
    cbar=plt.colorbar(sm,cax)
  1. 第一个命令通过将right = 1更改为right = 0.92,将左侧的四个子图调整到左侧,从而将彩条放置在其右侧。
  2. 第二个命令生成新的图形轴,以色条填充。数字对应于具有以下特征的矩形区域[左,下,宽度,高度],因此可以更改颜色条的宽度,位置和长度。
  3. 然后,第4条命令使用colormap cm绘制颜色条,将颜色映射到第3条命令所表示的子面板中所用的颜色,并使用cax轴。

这将导致以下绘图: Solution