python - 如何将色阶标签移动到 matplotlib/xarray 中彩色字段的中间？

Question

我有一个包含离散值(=类别/因素)的热图，创建如下:

import xarray
dats =  xarray.DataArray([[2,4,3,5,1],[8,4,5,3,2], [9,3,4,4,1]])
dats.plot(levels=range(0, 11, 1), cmap='Blues')

我想将色标/图例中的数字移动到相应字段的中间。我(和同事)发现他们在字段交叉处的位置有点误导，特别是考虑到热图本身中 x 轴的刻度位于字段的正中间。

如何将色标上的这些值移动到字段的中间(并截掉 10，这不是数据集中的值)？

Answer 1

为此，您必须直接使用 matplotlib:

import xarray
import matplotlib.pylab as plt
dats =  xarray.DataArray([[2,4,3,5,1],[8,4,5,3,2], [9,3,4,4,1]])
qm = dats.plot(levels=range(0, 11, 1), cmap='Blues', add_colorbar = False)

cbar = plt.colorbar(qm)
cbar.ax.get_yaxis().set_ticks([])
for j in range(0, 10, 1):
    cbar.ax.text(.5, (j+.5) / 10., j, ha='center', va='center', color='red')
cbar.ax.get_yaxis().labelpad = 15

plt.show()