下你所需,载你所想!
IT技术源码资料下载网站

使用seaborn也可以绘制热图

:其他软件 2020-10-08 01:02:28

使用seaborn也可以绘制热图

1. heatmap, 绘制普通的热图
2. clustermap,绘制带聚类数的热图
1. heatmap
相比matplotlib的imshow功能,该函数提供了更加简洁的接口,可以轻松实现文字注释的添加等功能,基本用法如下
>>> import numpy as np
>>> data = np.random.rand(10, 10)
>>> sns.heatmap(data)
>>> plt.show()
输出结果如下
在imshow中的部分参数在该函数中也是可以使用的,比如vmin, vmax,cmap等参数。除了通用参数外,该函数有两个特色,第一就是可以方便的添加分割线,使图片更加的美观,使用linescolor和linewidth参数指定分割线的颜色和宽度,用法如下
>>> sns.heatmap(data, linewidth=1)
>>> plt.show()
第二个特色是添加数字注释,在单元格上显示对应的数值,用法如下
>>> sns.heatmap(data, linewidth=1, annot=True)
>>> plt.show()
clustermap
clustermap绘制带聚类数的热图,基本用法如下
>>> data = np.random.rand(10,5)
>>> df = pd.DataFrame(data)
>>> df.columns = ['sampleA', 'sampleB', 'sampleC', 'sampleD', 'sampleE']
>>> sns.clustermap(df)
>>> plt.show()