關於比例和圖例的自定義熱圖問題

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。