Heat Map¶

En Matplotlib, se puede crear un mapa de calor ("heatmap") mediante el uso de la función imshow() para mostrar una matriz de valores como una imagen en color. El mapa de calor es una forma efectiva de visualizar datos 2D en una escala de colores.

Aquí hay un ejemplo de cómo crear un mapa de calor en Matplotlib en Python:

In [1]:

import matplotlib.pyplot as plt
import numpy as np

# Crear datos de ejemplo
np.random.seed(0)
data = np.random.randn(10, 10)

# Crear mapa de calor
plt.imshow(data, cmap='coolwarm')
plt.colorbar()
plt.title('Mapa de calor de ejemplo')
plt.show()

import matplotlib.pyplot as plt import numpy as np # Crear datos de ejemplo np.random.seed(0) data = np.random.randn(10, 10) # Crear mapa de calor plt.imshow(data, cmap='coolwarm') plt.colorbar() plt.title('Mapa de calor de ejemplo') plt.show()

En este ejemplo, se ha creado una matriz de 10x10 de datos de ejemplo utilizando la función randn() de NumPy. Luego, se ha utilizado la función imshow() para mostrar la matriz de datos como un mapa de calor en la escala de colores "coolwarm". Se ha agregado una barra de color con el comando colorbar() y se ha agregado un título con el comando title().

Veamos más ejemplos:

1. Mapa de calor con valores numéricos

In [6]:

import matplotlib.pyplot as plt
import numpy as np

# Crear datos de ejemplo
np.random.seed(0)
data = np.random.randn(10, 10)

# Crear mapa de calor con valores numéricos
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='coolwarm')
cbar = ax.figure.colorbar(im, ax=ax)
cbar.ax.set_ylabel('Valor numérico', rotation=-90, va="bottom")
ax.set_title('Mapa de calor con valores numéricos')
plt.show()

import matplotlib.pyplot as plt import numpy as np # Crear datos de ejemplo np.random.seed(0) data = np.random.randn(10, 10) # Crear mapa de calor con valores numéricos fig, ax = plt.subplots() im = ax.imshow(data, cmap='coolwarm') cbar = ax.figure.colorbar(im, ax=ax) cbar.ax.set_ylabel('Valor numérico', rotation=-90, va="bottom") ax.set_title('Mapa de calor con valores numéricos') plt.show()

2. Mapa de calor con anotaciones de texto

In [10]:

import matplotlib.pyplot as plt
import numpy as np

# Crear datos de ejemplo
np.random.seed(0)
data = np.random.randn(10, 10)

# Crear mapa de calor con anotaciones de texto
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='coolwarm')
cbar = ax.figure.colorbar(im, ax=ax)
cbar.ax.set_ylabel('Valor numérico', rotation=-90, va="bottom")
ax.set_title('Mapa de calor con anotaciones de texto')
for i in range(data.shape[0]):
    for j in range(data.shape[1]):
        text = ax.text(j, i, '{:.1f}'.format(data[i, j]),
                       ha="center", va="center", color="black")
plt.show()

import matplotlib.pyplot as plt import numpy as np # Crear datos de ejemplo np.random.seed(0) data = np.random.randn(10, 10) # Crear mapa de calor con anotaciones de texto fig, ax = plt.subplots() im = ax.imshow(data, cmap='coolwarm') cbar = ax.figure.colorbar(im, ax=ax) cbar.ax.set_ylabel('Valor numérico', rotation=-90, va="bottom") ax.set_title('Mapa de calor con anotaciones de texto') for i in range(data.shape[0]): for j in range(data.shape[1]): text = ax.text(j, i, '{:.1f}'.format(data[i, j]), ha="center", va="center", color="black") plt.show()