Crear un gráfico de barras agrupadas con Matplotlib

Introducción

En este laboratorio, aprenderá a crear un gráfico de barras agrupado con unidades utilizando Matplotlib. El gráfico de barras mostrará la altura de la taza por grupo y elección de bebida. El tutorial lo guiará a través de los pasos para crear el gráfico utilizando código de Python.

Consejos sobre la VM

Una vez que se haya completado la inicialización de la VM, haga clic en la esquina superior izquierda para cambiar a la pestaña Cuaderno y acceder a Jupyter Notebook para practicar.

A veces, es posible que tenga que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de operaciones no se puede automatizar debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Deje comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar las bibliotecas necesarias

El primer paso es importar las bibliotecas necesarias. Utilizaremos NumPy y Matplotlib para crear el gráfico de barras. Las unidades cm e inch se definen en el archivo basic_units.py, que se utiliza para convertir las unidades.

import matplotlib.pyplot as plt
import numpy as np
from basic_units import cm, inch

Definir los datos

El siguiente paso es definir los datos que utilizaremos para crear el gráfico de barras. Crearemos dos conjuntos de datos: uno para el té y otro para el café. Cada conjunto tendrá cinco valores, uno para cada grupo. La altura de las tazas se representará en centímetros.

N = 5
tea_means = [15*cm, 10*cm, 8*cm, 12*cm, 5*cm]
tea_std = [2*cm, 1*cm, 1*cm, 4*cm, 2*cm]
coffee_means = (14*cm, 19*cm, 7*cm, 5*cm, 10*cm)
coffee_std = (3*cm, 5*cm, 2*cm, 1*cm, 2*cm)

Crear los objetos Figura y Ejes

El siguiente paso es crear los objetos Figura y Ejes. Utilizaremos la función subplots() para crear los objetos Figura y Ejes.

fig, ax = plt.subplots()
ax.yaxis.set_units(inch)

Definir los parámetros del gráfico de barras

El siguiente paso es definir los parámetros para el gráfico de barras. Definiremos las ubicaciones x para los grupos, el ancho de las barras y las etiquetas para las marcas de los ejes x.

ind = np.arange(N)    ## las ubicaciones x para los grupos
width = 0.35         ## el ancho de las barras
ax.set_xticks(ind + width / 2, labels=['G1', 'G2', 'G3', 'G4', 'G5'])

Crear el gráfico de barras

El siguiente paso es crear el gráfico de barras. Utilizaremos la función bar() para crear el gráfico. Crearemos dos conjuntos de barras, una para el té y otra para el café. También agregaremos barras de error al gráfico.

ax.bar(ind, tea_means, width, bottom=0*cm, yerr=tea_std, label='Tea')
ax.bar(ind + width, coffee_means, width, bottom=0*cm, yerr=coffee_std,
       label='Coffee')

Agregar etiquetas y título al gráfico

El último paso es agregar etiquetas y un título al gráfico. Agregaremos un título al gráfico, una etiqueta para el eje x y una leyenda para el gráfico.

ax.set_title('Cup height by group and beverage choice')
ax.set_xlabel('Group')
ax.legend()
ax.autoscale_view()

Resumen

En este laboratorio, aprendiste cómo crear un gráfico de barras agrupadas con unidades utilizando Matplotlib. Aprendiste cómo importar las bibliotecas necesarias, definir los datos, crear los objetos Figura y Ejes, definir los parámetros del gráfico de barras, crear el gráfico de barras y agregar etiquetas y un título al gráfico. Con este conocimiento, puedes crear tus propios gráficos de barras utilizando Matplotlib.