python - 将刻度添加到 matplotlib colorbar

Question

在下面的简单代码中:

dat = np.linspace(0.1,0.9,4)*np.ones((4,1))
fig, ax = plt.subplots()
cax = ax.imshow(dat, interpolation='none')
cbar = fig.colorbar(cax)
plt.show()

给出:

我希望颜色条的范围为 [0,1]。这可以通过以下方式实现:

dat = np.linspace(0.1,0.9,4)*np.ones((4,1))
fig, ax = plt.subplots()
cax = ax.imshow(dat, interpolation='none')
cbar = fig.colorbar(cax)
cbar.set_clim(0, 1)
cbar.set_ticks(np.linspace(0,1,11))
plt.show()

结果是:

很明显，随着颜色变浅，范围发生了变化，但刻度没有变化，即 cbar.set_ticks() 没有按预期执行。如何更改刻度线？

Answer 1

我不知道为什么颜色条不调整，但你可以通过像这样更改对 imshow 的调用来实现你想要的

dat = np.linspace(0.1, 0.9, 4) * np.ones((4,1))
fig, ax = plt.subplots()
cax = ax.imshow(dat, interpolation='none', vmin=0, vmax=1)
cbar = fig.colorbar(cax)
plt.show()

注意额外的 vmin 和 vmax 参数。