从imshow获取数组而不绘制它

Question

我正在使用matplotlib.pyplot.imshow(img, cmap=cm.jet)。 img是2D矩阵。的 生成的图像带有颜色，所以我认为颜色图会为图像添加3个颜色通道。

如何在不首先保存图像或使用imshow绘制图像的情况下以numpy数组（包括3个颜色通道）的形式检索图像？

Answer 1

norm = plt.Normalize(img.min(), img.max())
numpy_array = cm.jet(norm(img))