Mapas de calor en Python

imshow dibuja una matriz; la barra de color y los ticks la hacen legible.

import matplotlib

matplotlib.use("Agg")
import matplotlib.pyplot as plt
import numpy as np

lenguajes = ["python", "rust", "sql"]
dias = ["lun", "mar", "mié"]
malla = np.array([[88, 97, 104], [80, 85, 91], [70, 76, 79]])

figura, ejes = plt.subplots()
imagen = ejes.imshow(malla, cmap="magma")
ejes.set_xticks(range(len(dias)), labels=dias)
ejes.set_yticks(range(len(lenguajes)), labels=lenguajes)
for fila in range(malla.shape[0]):
    for col in range(malla.shape[1]):
        ejes.text(col, fila, malla[fila, col], ha="center", va="center",
                  color="white", fontsize=8)
figura.colorbar(imagen, ax=ejes, label="tpm")
figura.savefig("heatmap.png")

Cómo funciona

1. imshow mapea valores a colores sobre una cuadrícula.
2. Define ambos juegos de ticks, o los ejes mienten sobre las etiquetas.
3. Escribir el número en cada celda le gana a adivinar por el color.

Palabras clave y builtins usados aquí

• as
• for
• len
• range

El intento, en números

Líneas

20

Caracteres a escribir

635

Tokens

224

Ritmo de tres estrellas

105 tpm

Al ritmo de tres estrellas de 105 tokens por minuto, este intento toma unos 128 segundos.

Escribe este fragmento

Paso 5 de 5 en Tipos de gráfica; paso 9 de 14 en Graficar con matplotlib.

← Anterior Siguiente →