python - 如何在 seaborn 热图中反转色标

Question

对于不同的值，seaborn 默认情况下似乎以暖色调(橙色)显示大数字，以冷色调(蓝色)显示小数字。

如果我需要将颜色切换为相反的颜色，以蓝色显示大数字而橙色显示小数字，该怎么做？

我已经搜索过了，但还没有找到方法。

sns.heatmap(flights, center=flights.loc["January", 1955])

Answer 1

您可以通过将 _r 附加到名称来反转所有 matplotlib 颜色图，即 plt.cm.coolwarm 与 plt.cm.coolwarm_r。

我相信 seaborn 默认使用 cubehelix 颜色图。

所以你会这样做:

from matplotlib import pyplot
import seaborn as sns

colormap = pyplot.cm.cubehelix_r
flights = sns.load_dataset('flights').pivot("month", "year", "passengers")
sns.heatmap(flights, cmap=colormap)