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 as mpl
import matplotlib.pyplot as plt
import matplotlib.colors as colors

上面的那种方式用于观看分析十分方便，但是图像的大小发生了变化，保存之后又白色的边距一直存在，如果只是想要把原图变成彩色的图像，尺寸大小不变，可以使用plt.imsave()

plt.imsave(image_dir,image_data,cmap=my_cmap)