Matplotlib 颜色图背景

Question

我正在开发一个基于颜色图生成图像的函数。这是我尝试过的：

#the variable 'name' is supplied before
gradient = np.vstack((np.linspace(0, 1, 256), np.linspace(0, 1, 256)))
plt.imshow(gradient, aspect='auto', cmap=plt.get_cmap(name))
plt.savefig("color.png", transparent=True)

但是，当我导出它时有一个透明边框：

当您在新标签页中打开它时，您将能够看到边框。有没有办法去掉这个边框？

Answer 1

您应该添加plt.axis('off') 以关闭轴框架，并添加plt.subplots_adjust(top = 1, right = 1, bottom = 0, left = 0) 以删除图像周围的垫。

完整代码

import numpy as np
import matplotlib.pyplot as plt

name = 'jet'
gradient = np.vstack((np.linspace(0, 1, 256), np.linspace(0, 1, 256)))
plt.imshow(gradient, aspect = 'auto', cmap = plt.get_cmap(name))
plt.axis('off')
plt.subplots_adjust(top = 1, right = 1, bottom = 0, left = 0)
plt.savefig("color.png", transparent = True)