Numpy、matplotlib实现二维数据到图像的转换，添加colormap，无边距显示

1.首先需要得到原始图像的大小

从向量shape到矩阵：

import numpy as np
data = np.reshape(vector,(101,101))
#101*101是要转换成的图片的大小

2.转换成灰度图

from PIL import Image
new_im = Image.fromarray(data.astype(np.uint8))

3.根据灰度值转成彩色图

4.自定义colormap

这个包中自带有很多colormap，如果没有特殊的要求，可以直接引用，具体系统自带的colormap在这里。下面介绍两种比较方便的方法：

5.无边距保存

需要导入的库：

import matplotlib