我有一个要在颜色栏中绘制的 2D numpy 数组。我无法更改轴以便它们显示我的数据集。垂直轴从 0 到 100 “向下”,而我希望它从 0.0 到 0.1 “向上”。所以我需要做两件事:
- 使用 np.flipud() 翻转数组,然后也“翻转”轴
- 将标签更改为从 0.0 到 0.1,而不是 0 到 100
这是我的颜色条图当前的示例:
这是代码:
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()
有人有什么建议吗?提前致谢!