matplotlib中使用imshow绘制二维图

这里所指的二维图,是二维矩阵数据的平面色彩显示

# -*- coding:utf-8 -*- from matplotlib import mpl import matplotlib.pyplot as plt import numpy as np #---------------------------------------------- data=np.clip(np.random.randn(5,5),-1,1) #生成随机数据,5行5列,最大值1,最小值-1 fig = plt.figure() # 第一个子图,按照默认配置 ax = fig.add_subplot(221) ax.imshow(data) # 第二个子图,使用自定义的colormap ax = fig.add_subplot(222) cmap=mpl.cm.cool #可以使用自定义的colormap ax.imshow(data,cmap=cmap) # 第三个子图增加一个colorbar ax = fig.add_subplot(223) cmap=mpl.cm.hot #可以使用自定义的colormap im=ax.imshow(data,cmap=cmap) plt.colorbar(im) ## 第四个子图可以调整colorbar ax = fig.add_subplot(224) cmap = mpl.cm.winter norm = mpl.colors.Normalize(vmin=-1, vmax=1) im=ax.imshow(data,cmap=cmap) plt.colorbar(im,cmap=cmap, norm=norm,ticks=[-1,0,1]) plt.show()