Criar Multiplot com Eixos Parasitas em Matplotlib

Beginner

This tutorial is from open-source community. Access the source code

Introdução

Neste laboratório, aprenderemos como criar um gráfico com múltiplos conjuntos de dados usando eixos parasitas em Python Matplotlib. Usaremos mpl_toolkits.axes_grid1.parasite_axes.host_subplot e mpl_toolkits.axisartist.axislines.Axes para criar um gráfico com múltiplos eixos y.

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 Bibliotecas

Primeiramente, precisamos importar as bibliotecas necessárias. Usaremos matplotlib.pyplot para plotar e mpl_toolkits para criar eixos parasitas (parasite axis).

import matplotlib.pyplot as plt
from mpl_toolkits import axisartist
from mpl_toolkits.axes_grid1 import host_subplot

Criar Eixos Host e Parasitas

Criaremos um eixo host (host axis) e dois eixos parasitas usando as funções host_subplot() e twinx(). A função host_subplot() cria um eixo host, e a função twinx() cria eixos parasitas que compartilham o mesmo eixo x com o eixo host.

host = host_subplot(111, axes_class=axisartist.Axes)
plt.subplots_adjust(right=0.75)

par1 = host.twinx()
par2 = host.twinx()

Ajustar Eixos Parasitas

Precisamos ajustar a posição dos eixos parasitas. A função new_fixed_axis() é usada para criar um novo eixo y no lado direito do gráfico. A função toggle() é usada para ativar todos os ticks e rótulos do eixo y direito.

par2.axis["right"] = par2.new_fixed_axis(loc="right", offset=(60, 0))

par1.axis["right"].toggle(all=True)
par2.axis["right"].toggle(all=True)

Plotar os Dados

Plotaremos três conjuntos de dados no mesmo gráfico: Densidade (Density), Temperatura (Temperature) e Velocidade (Velocity). Usaremos a função plot() para plotar os dados.

p1, = host.plot([0, 1, 2], [0, 1, 2], label="Density")
p2, = par1.plot([0, 1, 2], [0, 3, 2], label="Temperature")
p3, = par2.plot([0, 1, 2], [50, 30, 15], label="Velocity")

Definir Limites e Rótulos dos Eixos

Definiremos os limites e rótulos do eixo x e do eixo y para cada eixo usando a função set().

host.set(xlim=(0, 2), ylim=(0, 2), xlabel="Distance", ylabel="Density")
par1.set(ylim=(0, 4), ylabel="Temperature")
par2.set(ylim=(1, 65), ylabel="Velocity")

Adicionar Legenda e Cor

Adicionaremos uma legenda ao gráfico e coloriremos os rótulos de cada eixo para corresponder à cor do conjunto de dados correspondente, usando as funções legend() e label.set_color().

host.legend()

host.axis["left"].label.set_color(p1.get_color())
par1.axis["right"].label.set_color(p2.get_color())
par2.axis["right"].label.set_color(p3.get_color())

Exibir o Gráfico

Finalmente, exibiremos o gráfico usando a função show().

plt.show()

Resumo

Neste laboratório, aprendemos como criar um gráfico com múltiplos conjuntos de dados usando eixos parasitas (parasite axis) em Python Matplotlib. Usamos mpl_toolkits.axes_grid1.parasite_axes.host_subplot e mpl_toolkits.axisartist.axislines.Axes para criar um gráfico com múltiplos eixos y. Também aprendemos como ajustar a posição dos eixos parasitas, plotar os dados, definir os limites e rótulos dos eixos, adicionar legenda e cor ao gráfico, e exibir o gráfico.