Me gustaría hacer un mapa de calor como este (mostrado enFlowingData):

Los datos de origen sonaquí, pero los datos aleatorios y las etiquetas estarían bien de usar, es decir,

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

Hacer el mapa de calor es bastante fácil en matplotlib:

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

Y hasta encontré unmapa de colores Argumentos que parecen acertados:heatmap = plt.pcolor(data, cmap=matplotlib.cm.Blues)

Pero más allá de eso, no puedo imaginar cómo mostrar etiquetas para las columnas y filas y mostrar los datos en la orientación correcta (origen en la parte superior izquierda en lugar de en la parte inferior izquierda).

Intentos de manipularheatmap.axes (p.ej.heatmap.axes.set_xticklabels = column_labels) han fallado todos. ¿Que me estoy perdiendo aqui?