Eu gostaria de fazer um mapa de calor como este (mostrado naFlowingData):

Os dados de origem sãoAqui, mas os dados e rótulos aleatórios seriam bons para usar, ou seja,

import numpy
column_labels = list('ABCD')
row_labels = list('WXYZ')
data = numpy.random.rand(4,4)

Fazer o mapa de calor é bastante fácil no matplotlib:

from matplotlib import pyplot as plt
heatmap = plt.pcolor(data)

E eu até encontrei umcolormap argumentos que parecem corretos:heatmap = plt.pcolor(data, cmap=matplotlib.cm.Blues)

Mas, além disso, não consigo descobrir como exibir rótulos para as colunas e linhas e exibir os dados na orientação correta (origem no canto superior esquerdo em vez de canto inferior esquerdo).

Tentativas de manipularheatmap.axes (por exemplo.heatmap.axes.set_xticklabels = column_labels) todos falharam. O que estou perdendo aqui?