matplotlib中使用imshow绘制二维图

#coding:utf-8
from matplotlib import mpl
import matplotlib.pyplot as plt#载入matplotlib快速绘图的函数库
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)
ax.imshow(data, cmap = mpl.cm.cool)

#第三个子图，增加一个colorbar
ax = fig.add_subplot(223)
plt.colorbar(ax.imshow(data, cmap = mpl.cm.hot))

#第四个子图，可以调整colrobar
ax = fig.add_subplot(224)
plt.colorbar(ax.imshow(data, cmap = mpl.cm.winter), cmap = mpl.cm.winter, norm = mpl.colors.Normalize(vmin=-1,vmax=1), ticks=[-1,0,1])

plt.show()