关于比例和图例的自定义热图问题

Question

我尝试创建自定义热图，但在 y 尺度上遇到问题。

我构建的代码如下：

cmap = colors.ListedColormap(['red','green'])
bounds=[-1,1]
norm = colors.BoundaryNorm(bounds, cmap.N)
heatmap = plt.pcolor(np.array(result), cmap=cmap, norm=norm)
plt.colorbar(heatmap, ticks=[-1,0,1])

其中 result 是 3 列 6 行的数组，其中包含 -1 或 1 个值

我怎样才能编辑它以获得

0 1 2 3 

Answer 1

您可以修改刻度范围和频率：

plt.xticks(np.arange(0, 4, 1.0))

将从 0 到 4（不包括 4，所以直到 3）进行刻度，它们之间的间隔为 1。