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()