提问者:小点点

获取描述任意matplolib/seaborn绘图的numpy数组


是否可以获取描述任意matplolib/seaborn绘图的numpy数组,而不将其显式保存为图像文件并读取?

例如,假设我有一个附加的图像,它是一个sns。数组的kdeplot()。现在,我可以获得这个绘图的numpy数组,而不必先将其保存到文件并读取它吗?比如:

img = sns.kdeplot(arr)
img_arr = img.some_function() # Returns a numpy array describing the plot


共1个答案

匿名用户

fig.canvas。tostring_rgb()会有帮助:

fig, ax = plt.subplots()
sns.kdeplot(arr, ax=ax)

img_arr = np.fromstring(fig.canvas.tostring_rgb(), 
                        dtype=np.uint8,
                        sep='')