使用和不使用颜色条对齐matplotlib子绘图轴(用于2个以上的轴)

import matplotlib.pyplot as plt
import numpy as np

# 创建包含多个子图的图像对象
fig, axs = plt.subplots(2, 2, sharex=True, sharey=True)

# 绘制子图1
axs[0, 0].imshow(np.random.rand(10, 10), cmap='viridis')
axs[0, 0].set_title('Subplot 1')

# 绘制子图2
axs[0, 1].imshow(np.random.rand(10, 10), cmap='viridis')
axs[0, 1].set_title('Subplot 2')

# 绘制子图3
axs[1, 0].imshow(np.random.rand(10, 10), cmap='viridis')
axs[1, 0].set_title('Subplot 3')

# 绘制子图4
axs[1, 1].imshow(np.random.rand(10, 10), cmap='viridis')
axs[1, 1].set_title('Subplot 4')

# 添加颜色条
fig.colorbar(axs[0, 0].imshow(np.random.rand(10, 10), cmap='viridis'), ax=axs, location='right')

# 显示图像
plt.show()