如何自定义带有文本标签的海运热图中的色条

Question

我想为一个热图创建一个色条图例，而不是数字有文本。

例如，假设我要展示一些城市之间的人口分布情况。但在彩色栏中，我想要三个文本“低人口”、“中等”和“高”，而不是数字，它显示人口的范围。

这一点也不专业，但我处理这个问题的代码却受到了打击：

ax = sns.heatmap(Matrix_of_Population, cmap='viridis', cbar_kws={'label': 'Low      Medium    high'})

但是除了这些文本之外，人口的数量也是如此。

Answer 1

您可以在颜色栏中指定特定滴答的标签：

import seaborn as sns
import numpy as np 


a = np.random.randn(20,20)
ax = sns.heatmap(a, cmap='viridis')
c_bar = ax.collections[0].colorbar
c_bar.set_ticks([-2, 0, 2])
c_bar.set_ticklabels(['Low', 'Medium', 'High'])

​

​