位置: IT常识 - 正文

使用Python Seaborn绘制热力图（heatmap）的时候怎么改变配色(使用灭火器人要站在上风口还是下风口)

编辑：rootadmin

使用Python Seaborn绘制热力图（heatmap）的时候怎么改变配色

推荐整理分享使用Python Seaborn绘制热力图（heatmap）的时候怎么改变配色(使用灭火器人要站在上风口还是下风口)，希望有所帮助，仅作参考，欢迎阅读内容。

文章相关热门搜索词:使用新型病毒会增加艾滋病风险吗,使用灭火器扑救火灾时要对准火焰的什么部位喷射,使用灭火器扑救火灾时要对准火焰的什么部位喷射,使用新型病毒会增加艾滋病风险吗,使用灭火器扑救火灾时要对准火焰的什么部位喷射,使用男用避孕套中,下面哪项正确?,使用新型病毒会增加艾滋病风险吗,使用公筷公勺能减少幽门螺杆菌的感染风险,内容如对您有帮助，希望把文章链接给更多的朋友！

看到最近有些论文中会对Transformer encoder的attention weights进行可视化，通常会使用heatmap，我参考了一些博客，感觉已经总结得很详细了，例如这篇：python绘制热度图(heatmap)_黄思博呀的博客-CSDN博客_python heatmap

不过我觉得有一点说得不是很清楚，我看完之后还是不知道怎么可以修改配色，参考一下官方文档：seaborn.heatmap — seaborn 0.12.2 documentation

只是说可以用cmap来定义color space：

使用Python Seaborn绘制热力图（heatmap）的时候怎么改变配色(使用灭火器人要站在上风口还是下风口)

cmap matplotlib colormap name or object, or list of colors, optional

The mapping from data values to color space. If not provided, the default will depend on whether center is set.

但是完全不知道怎么用啊，具体来说，可以加上cmap这个参数，例如：

plot=sns.heatmap(p_array, cmap='Greens_r')

具体的色彩空间可以参考这里：Choosing Colormaps in Matplotlib — Matplotlib 3.6.3 documentation

其实如果我们故意输错这个参数的话，报错信息里会提示所有可能的参数：

supported values are 'Accent', 'Accent_r', 'Blues', 'Blues_r', 'BrBG', 'BrBG_r', 'BuGn', 'BuGn_r', 'BuPu', 'BuPu_r', 'CMRmap', 'CMRmap_r', 'Dark2', 'Dark2_r', 'GnBu', 'GnBu_r', 'Greens', 'Greens_r', 'Greys', 'Greys_r', 'OrRd', 'OrRd_r', 'Oranges', 'Oranges_r', 'PRGn', 'PRGn_r', 'Paired', 'Paired_r', 'Pastel1', 'Pastel1_r', 'Pastel2', 'Pastel2_r', 'PiYG', 'PiYG_r', 'PuBu', 'PuBuGn', 'PuBuGn_r', 'PuBu_r', 'PuOr', 'PuOr_r', 'PuRd', 'PuRd_r', 'Purples', 'Purples_r', 'RdBu', 'RdBu_r', 'RdGy', 'RdGy_r', 'RdPu', 'RdPu_r', 'RdYlBu', 'RdYlBu_r', 'RdYlGn', 'RdYlGn_r', 'Reds', 'Reds_r', 'Set1', 'Set1_r', 'Set2', 'Set2_r', 'Set3', 'Set3_r', 'Spectral', 'Spectral_r', 'Wistia', 'Wistia_r', 'YlGn', 'YlGnBu', 'YlGnBu_r', 'YlGn_r', 'YlOrBr', 'YlOrBr_r', 'YlOrRd', 'YlOrRd_r', 'afmhot', 'afmhot_r', 'autumn', 'autumn_r', 'binary', 'binary_r', 'bone', 'bone_r', 'brg', 'brg_r', 'bwr', 'bwr_r', 'cividis', 'cividis_r', 'cool', 'cool_r', 'coolwarm', 'coolwarm_r', 'copper', 'copper_r', 'crest', 'crest_r', 'cubehelix', 'cubehelix_r', 'flag', 'flag_r', 'flare', 'flare_r', 'gist_earth', 'gist_earth_r', 'gist_gray', 'gist_gray_r', 'gist_heat', 'gist_heat_r', 'gist_ncar', 'gist_ncar_r', 'gist_rainbow', 'gist_rainbow_r', 'gist_stern', 'gist_stern_r', 'gist_yarg', 'gist_yarg_r', 'gnuplot', 'gnuplot2', 'gnuplot2_r', 'gnuplot_r', 'gray', 'gray_r', 'hot', 'hot_r', 'hsv', 'hsv_r', 'icefire', 'icefire_r', 'inferno', 'inferno_r', 'jet', 'jet_r', 'magma', 'magma_r', 'mako', 'mako_r', 'nipy_spectral', 'nipy_spectral_r', 'ocean', 'ocean_r', 'pink', 'pink_r', 'plasma', 'plasma_r', 'prism', 'prism_r', 'rainbow', 'rainbow_r', 'rocket', 'rocket_r', 'seismic', 'seismic_r', 'spring', 'spring_r', 'summer', 'summer_r', 'tab10', 'tab10_r', 'tab20', 'tab20_r', 'tab20b', 'tab20b_r', 'tab20c', 'tab20c_r', 'terrain', 'terrain_r', 'turbo', 'turbo_r', 'twilight', 'twilight_r', 'twilight_shifted', 'twilight_shifted_r', 'viridis', 'viridis_r', 'vlag', 'vlag_r', 'winter', 'winter_r'