Question

我想将我的绘图的颜色条设置为自定义高度，不一定要匹配绘图的大小。事实上，我希望colorbar的高度加上它上面的标题以匹配图形的高度。

使用

ax3 = divider.append_axes('right', size='10%', pad=0.3)
cb = plt.colorbar(Q, cax=ax3, ticks=[0.0, 3.0, 6.0, 9.0, 12.0, 15.0], format='%.1f')

我设法有一个与绘图高度相同的颜色栏，这已被多次询问，现在我想缩小它。

在其他问题中提供了建议之后，我决定在使用add_axes获取最后一个绘图轴的位置后，明确地给色条赋予get_position自己的轴。这就是我想要做的。此示例中没有数据也没有颜色条，只是为了表明我没有得到预期的结果：

   from __future__ import unicode_literals
   import numpy as np
   from scipy.interpolate import griddata
   import matplotlib
   import matplotlib.pyplot as plt
   from matplotlib.patches import Circle
   from matplotlib.pylab import cm
   import matplotlib.colors as colors
   from mpl_toolkits.axes_grid1 import make_axes_locatable, axes_size

   matplotlib.rcParams.update({'font.size': 8})

   fig = plt.figure()
   fig.set_size_inches(6.3,6.3)

   ax1 = plt.subplot(111)
   divider = make_axes_locatable(ax1)
   ax2 = divider.append_axes('right', size='100%', pad=0.3)

   axes = [ax1, ax2]
   ltypes = ['dashed', 'solid']

   xi = np.linspace(-18.125, 18.125, 11)
   yi = np.linspace(0, 28, 9)
   xv, yv = np.meshgrid(xi, yi)

   xcOdd = 0.2
   zcOdd = 0.725
   xcEven = 0.6
   zcEven = 0.725

   maskRadius = 0.15

   for i in range(2):
       ax = axes[i]
       ax.set_xlabel('distance [m]')
       if i == 0:
           ax.set_ylabel('depth [m]')
       if i == 1:
           ax.set_yticklabels([])
       ax.invert_yaxis()
       ax.tick_params(direction='in')
       ax.set_aspect('equal')
       odd = Circle((xcOdd, zcOdd), .15, linewidth=1.2, color='k', fill=False)
       even = Circle((xcEven, zcEven), .15, linewidth=1.2, linestyle=ltypes[i], color='k', fill=False)

       vmax = 15.
       vmin =  0.
       norm = matplotlib.colors.Normalize(vmin,vmax, clip=False)

       color_map = matplotlib.colors.ListedColormap(plt.cm.Greys(np.linspace(0.25, 1, 5)), "name")

       ax.add_patch(odd)
       pad = 0.03
       width = 0.03

       pos = ax2.get_position()

       ax3 = fig.add_axes([pos.xmax + pad, pos.ymin, width, 0.7*(pos.ymax-pos.ymin) ])

       plt.savefig('prova-vect-paper-test-2.eps', format='eps')

为什么get_position返回错误的边界框？

Answer 1

您需要在从.get_position()获取实际位置之前绘制画布。这是因为由于相同的纵横比，轴在绘制时改变尺寸和位置。

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import matplotlib.colors as colors
from mpl_toolkits.axes_grid1 import make_axes_locatable

matplotlib.rcParams.update({'font.size': 8})

fig = plt.figure()
fig.set_size_inches(6.3,6.3)

ax1 = plt.subplot(111)
divider = make_axes_locatable(ax1)
ax2 = divider.append_axes('right', size='100%', pad=0.3)

axes = [ax1, ax2]


xi = np.linspace(-18.125, 18.125, 11)
yi = np.linspace(0, 28, 9)
xv, yv = np.meshgrid(xi, yi)


for i in range(2):
    ax = axes[i]
    ax.set_xlabel('distance [m]')
    if i == 0:
        ax.set_ylabel('depth [m]')
    if i == 1:
        ax.set_yticklabels([])
    ax.invert_yaxis()
    ax.tick_params(direction='in')
    ax.set_aspect('equal')
    vmax = 15.
    vmin =  0.
    norm = colors.Normalize(vmin,vmax, clip=False)

    color_map = colors.ListedColormap(plt.cm.Greys(np.linspace(0.25, 1, 5)), "name")

    im = ax.imshow(yv, cmap=color_map, norm=norm)

pad = 0.03
width = 0.03

fig.canvas.draw()
pos = ax2.get_position()

ax3 = fig.add_axes([pos.xmax + pad, pos.ymin, width, 0.7*(pos.ymax-pos.ymin) ])
fig.colorbar(im, cax=ax3)

plt.show()

将matplotlib颜色条更改为自定义高度

1 个答案: