matplotlib绘制热力图

以在sklearn的svm为例
matplotlib绘制热力图_第1张图片

plt.figure(figsize=(8, 6))
plt.subplots_adjust(left=.2, right=0.95, bottom=0.15, top=0.95)
#设置标注前后左右的距离
plt.imshow(scores, interpolation='nearest', cmap=plt.cm.hot,
        norm=MidpointNormalize(vmin=0.2, midpoint=0.92))
plt.xlabel('gamma')
plt.ylabel('C')
plt.colorbar()
plt.xticks(np.arange(len(gamma_range)), gamma_range, rotation=45)
plt.yticks(np.arange(len(C_range)), C_range)
plt.title('Validation accuracy')
plt.show()

imshow(X, cmap=None, norm=None, aspect=None, interpolation=None, alpha=None, vmin=None, vmax=None, origin=None, extent=None, shape=None, filternorm=1, filterrad=4.0, imlim=None, resample=None, url=None, hold=None, data=None, **kwargs)

简单例子:

from matplotlib import pyplot as plt  

X = [[1,2],[3,4],[5,6]]  
plt.imshow(X)  
plt.colorbar()
plt.show() 

matplotlib绘制热力图_第2张图片
plt.colorbar(cax=None,ax=None,shrink=0.5)可设置Bar为一半长度。

Colormap:参数cmap用于设置热图的Colormap。(参考百度百科)
Colormap是MATLAB里面用来设定和获取当前色图的函数,可以设置如下色图:
hot 从黑平滑过度到红、橙色和黄色的背景色,然后到白色。
cool 包含青绿色和品红色的阴影色。从青绿色平滑变化到品红色。
gray 返回线性灰度色图。
bone 具有较高的蓝色成分的灰度色图。该色图用于对灰度图添加电子的视图。
white 全白的单色色图。
spring 包含品红和黄的阴影颜色。
summer 包含绿和黄的阴影颜色。
autumn 从红色平滑变化到橙色,然后到黄色。
winter 包含蓝和绿的阴影色。

interpolation=’nearest’是把相邻的相同的颜色连成片

参考资料:
1、https://jingyan.baidu.com/article/48a42057fe9895a92425048c.html
2、https://stackoverflow.com/questions/22121239/matplotlib-imshow-default-colour-normalisation
3、https://blog.csdn.net/goldxwang/article/details/76855200

你可能感兴趣的:(可视化)