Trazar un mapa de calor 2D con Matplotlib

| | | | | | |

Usando Matplotlib, quiero trazar un mapa de calor 2D. Mis datos son una matriz Numpy n por n, cada una con un valor entre 0 y 1. Entonces, para el elemento (i, j) de esta matriz, quiero trazar un cuadrado en la coordenada (i, j) en mi mapa de calor, cuyo color es proporcional al valor del elemento en la matriz.

¿Cómo puedo hacer esto?