以下是我绘制热图的方法:
import matplotlib.pyplt as plt
ax = plt.gca()
im = ax.imshow(values)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size='5%', pad=0.05)
plt.colorbar(im, cax=cax)
现在我想创建一个2×2子图,有4个不同的热图,并且都具有相同的热棒.我对如何实现这一目标完全不了解,并希望能够朝着正确的方向努力.
解决方法:
您可以使用AxesGrid
from mpl_toolkits.axes_grid1
执行此操作.请参阅示例here(具体来说,请查看该示例中的函数demo_grid_with_single_cbar).
我稍微修改了这个例子,对于你的情况是2×2网格,右边是彩色条.
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import AxesGrid
# Some random data
values1 = np.random.rand(10,10)
values2 = np.random.rand(10,10)
values3 = np.random.rand(10,10)
values4 = np.random.rand(10,10)
vals = [values1,values2,values3,values4]
fig = plt.figure()
grid = AxesGrid(fig, 111,
nrows_ncols=(2, 2),
axes_pad=0.05,
share_all=True,
label_mode="L",
cbar_location="right",
cbar_mode="single",
)
for val, ax in zip(vals,grid):
im = ax.imshow(val, vmin=0, vmax=1)
grid.cbar_axes[0].colorbar(im)
for cax in grid.cbar_axes:
cax.toggle_label(False)
plt.show()