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() 

이 방법은, 도면에서의 축 위치에 일단이 변화, 예를 들어 의존 피규어가 rezised 때문에, unforseable 일이 일어날 수 있습니다.

다른 방법은 그려진 좌표에 의존하지 않고 삽입 된 축을 사용하여 축 외부에 삽입하는 것입니다. 이 방법으로 인세 트가 위치한 좌표는 축 좌표이므로 색상 막대는 축에 따라 위치가 변경됩니다.

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)