我有一个2D的numpy数组,想在colorbar中绘制它。我遇到了一些问题,无法更改坐标轴以显示我的数据集。垂直轴从0到100向下延伸,而我希望它从0.0到0.1向上延伸。所以我需要做两件事:
以下是代码:
有没有人有建议呢?提前谢谢!
- 使用np.flipud()翻转数组,然后也要“翻转”轴
- 将标签更改为从0到0.1,而不是从0到100
![Example of Colorbar plot](https://istack.dev59.com/3YQvl.webp)
data = np.load('scorr.npy')
(x,y) = np.unravel_index(data.argmax(), data.shape)
max=data[x][y]
fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.imshow(data, interpolation='nearest')
cbar = fig.colorbar(cax, ticks=[-max, 0, max])
cbar.ax.set_yticklabels([str(-max), '0', str(max)])
plt.show()
有没有人有建议呢?提前谢谢!