python - 将刻度线置于热图行的中心

Question

我试图将下面热图上的 yticks 移动到它对应的行的中心。

这已经起作用了，除了我现在在图表底部看到了空白区域。有没有办法实现中间的刻度没有空白？

原始图像和代码:

heatmap = plt.pcolor(df2.T, cmap=cmap,  vmax=1, vmin=0)
plt.yticks(range(len(df2.columns )+1), o, size=15)

新图像和代码

heatmap = plt.pcolor(df2.T, cmap=cmap,  vmax=1, vmin=0)
plt.yticks(np.arange(0.5, len(df2.columns)+1,1), df2.columns, size=15)

尝试

我尝试过使用

plt.yticks(np.arange(0.5, len(df2.columns)+0.5,1), o, size=15

请注意从 +1 到 +0.5 的变化但是，这将我的底行减半。见下文:

Answer 1

您可以使用ylim设置y轴范围:

plt.ylim(0, df2.shape[1])