将Rainbow矩阵转换为RBG并另存为PNG

时间:2019-01-10 18:50:40

标签: python pandas numpy

我正在遵循本教程:https://pyts.readthedocs.io/en/latest/auto_examples/plot_gaf.html,以便生成GADF和GASF矩阵。

一切正常。 我只需要将X_gasf[0]中包含的结果另存为png,但我不知道该怎么做。

我已经尝试过:

fig = plt.imshow(X_gadf[0], cmap='rainbow', aspect='equal')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.axis('off')
plt.savefig(GADF_path + filename, pad_inches=0, bbox_inches='tight')

此代码有效,但是在图像周围仍然给我一个白色边框(对我来说,没有任何白色边框非常重要)。

我也尝试过:

# Test 1
matplotlib.image.imsave(filename, X_gadf[0])

# Test 2
im = Image.fromarray(X_gadf[0], mode='HSV')
im.save(GADF_path + filename)

# Test 3
cv2.imread(X_gadf[0])
imsave(filename, X_gadf[0])

但是这些解决方案都不起作用。看来X_gadf[0]是我无法管理的奇怪图像。

有什么想法吗?

1 个答案:

答案 0 :(得分:1)

您尝试过吗?

matplotlib.image.imsave(filename, X_gasf[0], cmap='rainbow')

在查看文档时,您似乎忘记了cmap='rainbow'