Изменение осей на графике цветовой шкалы matplotlib 2D-массива

У меня есть двумерный массив numpy, который я хочу отобразить на цветной панели. У меня проблемы с изменением оси, чтобы они отображали мой набор данных. Вертикальная ось идет «вниз» от 0 до 100, тогда как я хочу, чтобы она пошла «вверх» от 0,0 до 0,1. Поэтому мне нужно сделать две вещи:

  • Отразить массив с помощью np.flipud (), а затем «перевернуть» ось
  • Измените метки, чтобы они изменились с 0,0 на 0,1 вместо 0 на 100

Вот пример того, как сейчас выглядит мой график цветовой шкалы: Example of Colorbar plot

А вот код:

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()

Есть ли у кого-нибудь предложения? Заранее спасибо!

7
задан Doa 7 June 2011 в 14:44
поделиться