Este tipo de gráficas pueden ir acompañadas de una barra de color que muestre los valores correspondientes a cada uno de los colores usados, lo que puede facilitar enormemente la interpretación del gráfico. La función que genera esta barra es matplotlib.pyplot.colorbar, y no necesita argumentos adicionales. Por el contrario, el método de una figura colorbar requiere recibir como argumento el gráfico al que va asociado (véanse ejemplos más adelante).
Veamos un primer ejemplo creando una barra de color para una gráfica de curvas nivel creada con la función contour. Comenzamos recreando los valores X, Y y Z:
def f(x, y):
return np.sin(x) ** 2 + np.cos(5 + x * y) + x
x = np.linspace(0, 5, 100)
y = np.linspace(0, 5, 100)
X, Y = np.meshgrid(x, y)
Z = f(X, Y)
A continuación, generamos la gráfica y añadimos la función colorbar:
En el caso de asociarse la barra de color a una gráfica de curvas de nivel, la ayuda que supone es relativa. Pero si se asocia a una gráfica de curvas de nivel rellena, la mejora es sustancial. En el siguiente ejemplo se utiliza una de estas gráficas. En este caso vamos a usar el método colorbar asociado a la figura. Obsérvese que es necesario incluir como argumento una referencia a la gráfica generada:
La barra puede ser mostrada horizontalmente usando el parámetro orientation (en el siguiente ejemplo se usa el mapa de colores Reds)
El parámetro fraction controla el porcentaje de la figura que se dedicará a la barra de color, y el parámetro aspect controla el aspecto de la barra (la proporción alto-ancho). Jugando con ambos valores podemos mostrar barras más anchas que cubran la altura completa de la figura:
