如何为具有对数刻度的seaborn热图以纯数字显示刻度?

时间:2021-01-11 19:33:46

标签: python matplotlib seaborn

我正在使用具有对数刻度的 seaborn 生成热图。如何将颜色条标签从科学记数法更改为普通数字。

import math
from matplotlib.colors import LogNorm
vmax=2
vmin=0.5
center = (vmax+vmin)/2
log_norm = LogNorm(vmin=vmin, vmax=vmin)
cbar_ticks = [0.5, 0.66, 1, 1.5, 2]
ax = sns.heatmap(corr, square=True, mask=mask, cmap=cmap_type, linewidths=.5, vmax=vmax, vmin=vmin, norm=log_norm, cbar_kws={"ticks": cbar_ticks}, center=center)

generated graph

编辑:

以下代码解决了科学记数法的问题:

    import matplotlib.ticker as tkr
    formatter = tkr.ScalarFormatter(useMathText=True)
    formatter.set_scientific(False)
    cbar_kws={"ticks": cbar_ticks, "format": formatter}

norm=log_norm 正在添加额外的刻度(请参见下图)。如何删除这些?

Updated graph

2 个答案:

答案 0 :(得分:0)

seaborn 图是围绕 matplotlib 的包装,所以我建议这样做

ax.ticklabel_format(style='plain')

答案 1 :(得分:0)

以下代码有帮助。

对于问题的第一部分,在 cbar_kws 参数中添加“格式”有帮助。 (https://stackoverflow.com/a/35419495/8881141):

import matplotlib.ticker as tkr
formatter = tkr.ScalarFormatter(useMathText=True)
formatter.set_scientific(False)

ax = sns.heatmap(corr, square=True, mask=mask, cmap=cmap_type, linewidths=.5, vmax=vmax, vmin=vmin, norm=log_norm, cbar_kws={"ticks": cbar_ticks, "format": formatter}, center=center)

对于由 matplotlib 中的错误引起的第二部分,添加以下内容将删除次要刻度。 (https://stackoverflow.com/a/65676007/11897903):

ax.collections[0].colorbar.ax.yaxis.set_ticks([], minor=True)