Matplotlib matshow labels

Я начал использовать matplotlib месяц назад, поэтому я все еще участвую. Я пытаюсь сделать тепловую карту с matshow. Мой код следующий:

data = numpy.array(a).reshape(4, 4)  
cax = ax.matshow(data, interpolation='nearest', cmap=cm.get_cmap('PuBu'), norm=LogNorm())  
cbar = fig.colorbar(cax)

ax.set_xticklabels(alpha)  
ax.set_yticklabels(alpha)

где alpha - это модель из django с 4 полями: 'ABC', 'DEF', 'GHI', 'JKL'

Дело в том, что я не знаю, почему, ярлык "ABC" не появляется, оставляя последнюю ячейку без метки.
Если бы кто-нибудь понял, как изменить мой script таким образом, чтобы он появился в "ABC", я был бы благодарен:)

ОТВЕТЫ

Ответ 1

Что происходит, так это то, что xticks фактически выходят за пределы отображаемой фигуры при использовании matshow. (Я не совсем уверен, почему это так. Я почти никогда не использовал matshow.)

Чтобы продемонстрировать это, посмотрите на вывод ax.get_xticks(). В вашем случае это array([-1., 0., 1., 2., 3., 4.]). Поэтому, когда вы устанавливаете метки xtick, "ABC" имеет значение & lt; -1, -1> и не отображается на рисунке.

Самое простое решение - просто добавить пустую метку в список меток, например,

ax.set_xticklabels(['']+alpha)
ax.set_yticklabels(['']+alpha)

В качестве полного примера:

import numpy as np
import matplotlib.pyplot as plt

alpha = ['ABC', 'DEF', 'GHI', 'JKL']

data = np.random.random((4,4))

fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(data, interpolation='nearest')
fig.colorbar(cax)

ax.set_xticklabels(['']+alpha)
ax.set_yticklabels(['']+alpha)

plt.show()

Matshow example