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()