python使用seaborn绘制热力图

最近写论文，需要把特征图以热图的形式展示出来，因此写下这篇笔记：

需要用到的库为seaborn

具体代码如下：

import seanborn as sns
sns.set()
# 注意： your_data只能是二维矩阵，特征图为多通道时可以使用mean函数取通道上的平均。
ax = sns.heatmap(your_data, cmap="rainbow", xticklabels=False, yticklabels=False, cbar=True)
b = ax.get_figure()
b.savefig(your_save_path)

seaborn库是一个可视化数据的库，heatmap是该库中的一个用来绘制热力图的函数。

heatmap()函数中有很多参数，可以根据自己的绘图需要进行选择，以下是该函数的完整参数。

seaborn.heatmap(data, vmin=None, vmax=None, cmap=None, center=None, robust=False, annot=None, fmt='.2g', annot_kws=None, linewidths=0, linecolor='white', cbar=True, cbar_kws=None, cbar_ax=None, square=False, xticklabels='auto', yticklabels='auto',