matplotlib中使用imshow绘制二维图

时间:2022-10-23 23:36:35
#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()

matplotlib中使用imshow绘制二维图