热图在python中的灰度图像之上

Question

简单明了的问题，我需要在python中的灰度图像上绘制热图，如下图所示。

我来自MATLAB。在那里，我将简单地生成概率图，通过颜色图传递它并将结果插入灰色标量图片。

我可以在python中做同样的事情，虽然我觉得不是最“pythonic”的方式（numpy）。您知道处理这种可视化的任何特定包吗？