matplotlib axesgrid - 额外的颜色条？

Question

我想添加另一个颜色条到我使用AxesGrid工具包的情节。例如，我在左边添加一个使用ImageGrid的彩条轴，然后手动在右边添加另一个。这里有一个简单的例子：

f = plt.figure(1) 
grid = ImageGrid(f, 111, # similar to subplot(111) 
       nrows_ncols=(2, 2), 
       axes_pad=0.01, 
       add_all=True, 
       cbar_location="left", 
       label_mode='L', 
       cbar_mode="edge", 
       cbar_size="3%", 
       cbar_pad="2%", 
       ) 
for i in range(3): 
    m = grid[i].matshow(np.arange(100).reshape((10, 10))) 
plt.colorbar(m, grid.cbar_axes[0]) 
m = grid[3].matshow(np.arange(100).reshape((10, 10)), cmap='plasma') 
plt.colorbar(m, shrink=0.5, anchor=(0, 0)) 
plt.show() 

如何使新的颜色条与网格中其中一个子图的位置完全匹配？我至少设法使用缩小和锚点来固定尺寸和y位置......但是如果我尝试考虑子图之间的填充，并且它们是矩形而不是方形，它也会变得有点复杂......

Answer 1

一种选择是根据其中一个轴的位置手动放置色条轴。为此，首先需要绘制画布，以便知道位置。然后可以根据图像坐标创建一个新的坐标轴。这个新的轴将作为彩条轴。

import matplotlib.pyplot as plt 
import numpy as np 
from mpl_toolkits.axes_grid1 import ImageGrid 

fig = plt.figure(1) 
grid = ImageGrid(fig, 111, # similar to subplot(111) 
       nrows_ncols=(2, 2), 
       axes_pad=0.01, 
       add_all=True, 
       cbar_location="left", 
       label_mode='L', 
       cbar_mode="edge", 
       cbar_size="3%", 
       cbar_pad="2%", 
       ) 
for i in range(3): 
    m = grid[i].matshow(np.arange(100).reshape((10, 10))) 
plt.colorbar(m, grid.cbar_axes[0]) 
m = grid[3].matshow(np.arange(100).reshape((10, 10)), cmap='plasma') 

# first draw the figure, such that the axes are positionned 
fig.canvas.draw() 
#create new axes according to coordinates of image plot 
trans = fig.transFigure.inverted() 
g3 =grid[3].bbox.transformed(trans) 
pos = [g3.x1 + g3.width*0.02, g3.y0, g3.width*0.03, g3.height ] 
cax = fig.add_axes(pos) #l,b,w,h 
# add colorbar to new axes 
plt.colorbar(m, cax=cax) 


plt.show() 

此方法依赖于轴线的在图中的位置，一旦改变，例如因为这个数字被重新调整，所以不可能发生的事情可能会发生。

另一种不依赖绘制坐标的方法是（错误地）使用插入轴并将插入点放置在轴的外部。通过这种方式，插图所在的坐标就是坐标轴，所以颜色条会根据坐标轴改变其位置。

from mpl_toolkits.axes_grid1.inset_locator import inset_axes 

cax = inset_axes(grid[3], "3%", "100%", loc=3, bbox_to_anchor=(1.02,0,1,1), 
       bbox_transform=grid[3].transAxes, borderpad=0.0) 
plt.colorbar(m, cax=cax)