matplotlib绘制热力图

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/jin_tmac/article/details/82184457
以在sklearn的svm为例


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()
1
2
3
4
5
6
7
8
9
10
11
12
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()
1
2
3
4
5
6

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
————————————————
版权声明:本文为CSDN博主「jin_tmac」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/jin_tmac/article/details/82184457

原文地址:https://www.cnblogs.com/jfdwd/p/11425503.html