使用颜色图将二维数组转换为图像

Question

我使用小波 CWT 将信号解码为浮点值介于 -450.0 和 550.0 之间的二维数组，并且可以使用 matplotlib 的 imshow 函数显示它：

data=pywt.cwt(signal, scales, wavelet = wavelet)
plt.imshow(data, cmap = 'coolwarm', aspect = 'auto')
plt.show()

哪个工作得很好：

现在我想将此图像转换为数组，以便将其提供给 CNN 模型，但我找不到任何解决方案。有人能指出 e 的正确方向吗？