Mostrar barra de colores en Matplotlib

En este tutorial, discutiremos cómo mostrar una barra de colores en una figura de matplotlib en Python.

Para crear una barra de colores tenemos que usar la función matplotlib.pyplot.colorbar().

El siguiente código muestra un ejemplo simple de esto.

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

s = plt.scatter(s_x, s_y, c=s_x, cmap="viridis")

c = plt.colorbar()

En el ejemplo anterior, creamos una barra de colores simple colocada fuera del gráfico. Especificamos el mapa de colores usando el parámetro cmap.

También podemos especificar los ejes en los que deseamos mostrar la barra de colores. Si lo deseamos, podemos agregarlo sobre los ejes del gráfico.

Por ejemplo,

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

fig, ax = plt.subplots()
cax = fig.add_axes([0.27, 0.8, 0.5, 0.05])

im = ax.scatter(s_x, s_y, c=s_x, cmap="viridis")
fig.colorbar(im, cax=cax, orientation="horizontal")

En el ejemplo anterior, usamos la función subplots() para obtener la figura y los objetos de los ejes y usarla para crear los ejes para la barra de colores. Especificamos esto usando el parámetro cax en la función colorbar().

Además, tenga en cuenta el uso del parámetro orientation que alteró la orientación de la barra de color final. Aparte de esto, podemos usar diferentes parámetros para controlar la forma y el tamaño de la barra de colores. Por ejemplo, shrink puede reducir el tamaño de la barra de colores en un pequeño margen, aspect, que es la proporción de los lados de la barra, y hay muchos más.