Criar Gráfico de Barras Agrupadas com Matplotlib

Introdução

Neste laboratório, você aprenderá como criar um gráfico de barras agrupadas com unidades usando Matplotlib. O gráfico de barras exibirá a altura da xícara por grupo e a escolha da bebida. O tutorial irá guiá-lo pelos passos para criar o gráfico usando código Python.

Dicas da VM

Após a inicialização da VM, clique no canto superior esquerdo para mudar para a aba Notebook e acessar o Jupyter Notebook para praticar.

Às vezes, pode ser necessário aguardar alguns segundos para que o Jupyter Notebook termine de carregar. A validação das operações não pode ser automatizada devido a limitações no Jupyter Notebook.

Se você enfrentar problemas durante o aprendizado, sinta-se à vontade para perguntar ao Labby. Forneça feedback após a sessão, e resolveremos o problema prontamente para você.

Importar as Bibliotecas Necessárias

O primeiro passo é importar as bibliotecas necessárias. Usaremos NumPy e Matplotlib para criar o gráfico de barras. As unidades cm e inch são definidas no arquivo basic_units.py, que é usado para converter as unidades.

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

Definir os Dados

O próximo passo é definir os dados que usaremos para criar o gráfico de barras. Criaremos dois conjuntos de dados - um para chá e outro para café. Cada conjunto terá cinco valores, um para cada grupo. A altura das xícaras será representada em 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)

Criar os Objetos Figura e Eixos

O próximo passo é criar os objetos figura e eixos. Usaremos a função subplots() para criar os objetos figura e eixos.

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

Definir os Parâmetros do Gráfico de Barras

O próximo passo é definir os parâmetros para o gráfico de barras. Definiremos as localizações x para os grupos, a largura das barras e os rótulos para os x-ticks.

ind = np.arange(N)    ## the x locations for the groups
width = 0.35         ## the width of the bars
ax.set_xticks(ind + width / 2, labels=['G1', 'G2', 'G3', 'G4', 'G5'])

Criar o Gráfico de Barras

O próximo passo é criar o gráfico de barras. Usaremos a função bar() para criar o gráfico. Criaremos dois conjuntos de barras, um para chá e outro para café. Também adicionaremos barras de erro ao 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')

Adicionar Rótulos e Título ao Gráfico

O passo final é adicionar rótulos e um título ao gráfico. Adicionaremos um título ao gráfico, um rótulo para o eixo x e uma legenda para o gráfico.

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

Resumo

Neste laboratório, você aprendeu como criar um gráfico de barras agrupadas com unidades usando Matplotlib. Você aprendeu como importar as bibliotecas necessárias, definir os dados, criar os objetos de figura e eixos, definir os parâmetros do gráfico de barras, criar o gráfico de barras e adicionar rótulos e um título ao gráfico. Com este conhecimento, você pode criar seus próprios gráficos de barras usando Matplotlib.