Python Matplotlib | Tutorial Gráfico Bar of Pie

Introdução

Neste tutorial, você aprenderá como criar um gráfico "Bar of Pie" usando a biblioteca Matplotlib do Python. Um gráfico Bar of Pie é uma combinação de um gráfico de pizza (Pie chart) e um gráfico de barras empilhadas (stacked bar chart), onde a primeira fatia da pizza é "explodida" em um gráfico de barras com uma análise mais detalhada de suas características. Este gráfico é útil quando você deseja mostrar a distribuição de um conjunto de dados inteiro, ao mesmo tempo em que destaca categorias específicas.

Dicas para a 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

Antes de começarmos a criar o gráfico, precisamos importar as bibliotecas necessárias. Neste caso, usaremos matplotlib.pyplot e numpy.

import matplotlib.pyplot as plt
import numpy as np

Definir os dados para o gráfico

Em seguida, precisamos definir os dados que usaremos para criar o gráfico. Neste caso, usaremos os seguintes dados:

## pie chart parameters
overall_ratios = [.27, .56, .17]
labels = ['Approve', 'Disapprove', 'Undecided']
explode = [0.1, 0, 0]

## bar chart parameters
age_ratios = [.33, .54, .07, .06]
age_labels = ['Under 35', '35-49', '50-65', 'Over 65']

Criar o gráfico de pizza

Agora podemos criar o gráfico de pizza. Começamos definindo os objetos de figura e eixo:

## make figure and assign axis objects
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(9, 5))
fig.subplots_adjust(wspace=0)

Em seguida, definimos os parâmetros para o gráfico de pizza e o plotamos:

## rotate so that first wedge is split by the x-axis
angle = -180 * overall_ratios[0]
wedges, *_ = ax1.pie(overall_ratios, autopct='%1.1f%%', startangle=angle,
                     labels=labels, explode=explode)

Criar o gráfico de barras

Em seguida, criamos o gráfico de barras empilhadas. Começamos definindo os parâmetros para o gráfico:

## bar chart parameters
bottom = 1
width = .2

## Adding from the top matches the legend.
for j, (height, label) in enumerate(reversed([*zip(age_ratios, age_labels)])):
    bottom -= height
    bc = ax2.bar(0, height, width, bottom=bottom, color='C0', label=label,
                 alpha=0.1 + 0.25 * j)
    ax2.bar_label(bc, labels=[f"{height:.0%}"], label_type='center')

Conectar o gráfico de pizza e o gráfico de barras

Finalmente, conectamos o gráfico de pizza e o gráfico de barras usando ConnectionPatch:

## use ConnectionPatch to draw lines between the two plots
theta1, theta2 = wedges[0].theta1, wedges[0].theta2
center, r = wedges[0].center, wedges[0].r
bar_height = sum(age_ratios)

## draw top connecting line
x = r * np.cos(np.pi / 180 * theta2) + center[0]
y = r * np.sin(np.pi / 180 * theta2) + center[1]
con = ConnectionPatch(xyA=(-width / 2, bar_height), coordsA=ax2.transData,
                      xyB=(x, y), coordsB=ax1.transData)
con.set_color([0, 0, 0])
con.set_linewidth(4)
ax2.add_artist(con)

## draw bottom connecting line
x = r * np.cos(np.pi / 180 * theta1) + center[0]
y = r * np.sin(np.pi / 180 * theta1) + center[1]
con = ConnectionPatch(xyA=(-width / 2, 0), coordsA=ax2.transData,
                      xyB=(x, y), coordsB=ax1.transData)
con.set_color([0, 0, 0])
ax2.add_artist(con)
con.set_linewidth(4)

Exibir o gráfico

Finalmente, exibimos o gráfico:

plt.show()

Resumo

Neste tutorial, você aprendeu como criar um gráfico "Bar of Pie" usando a biblioteca Matplotlib do Python. Um gráfico Bar of Pie é útil quando você deseja mostrar a distribuição de um conjunto de dados inteiro, ao mesmo tempo em que destaca categorias específicas. Você também aprendeu como usar ConnectionPatch para conectar o gráfico de pizza e o gráfico de barras.

Criando um Gráfico Bar of Pie