在matplotlib中跨多个图执行颜色一致性

Question

我想知道是否有一种方法可以在灰度图像之间实现色彩一致性

例如，我首先生成一些玩具面具：

plots = []
for i in [1, 126, 255]:
    x = np.zeros((3, 3))
    x[1, 1] = i
    plots.append(x)

当我绘制它们时，我得到类似的东西：

但是理想情况下，我希望第一张图像接近黑色（因为它的值只有1），第二张图像是灰色，第三张图像是白色。

我可以解决的唯一方法是在绘制时在图像中的某处添加虚拟像素。例如。设置mask[0][0]=1，mask[0][1]=2等。有其他替代方法吗？