Como hacer un mapa de calor en Python

Necesito saber como se hace un mapa de calor en python, uso la librería
Import matplotlib
Import matplotlib. Pyplot as plt