Visualização com ConnectionPatch em Matplotlib

Beginner

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

Introdução

Matplotlib é uma poderosa biblioteca de visualização de dados em Python. Ela oferece uma variedade de ferramentas de visualização para criar gráficos e diagramas impressionantes. Neste laboratório, aprenderemos sobre ConnectionPatch, que é usado para desenhar uma linha entre pontos definidos em diferentes sistemas de coordenadas e/ou eixos.

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, vamos importar as bibliotecas necessárias.

import matplotlib.pyplot as plt
from matplotlib.patches import ConnectionPatch

Criar o gráfico

Em seguida, vamos criar o gráfico com dois subplots.

fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(6, 3))

Desenhar uma seta simples

Agora, vamos desenhar uma seta simples entre dois pontos em coordenadas de eixos dentro de um único eixo.

xyA = (0.2, 0.2)
xyB = (0.8, 0.8)
coordsA = "data"
coordsB = "data"
con = ConnectionPatch(xyA, xyB, coordsA, coordsB,
                      arrowstyle="-|>", shrinkA=5, shrinkB=5,
                      mutation_scale=20, fc="w")
ax1.plot([xyA[0], xyB[0]], [xyA[1], xyB[1]], "o")
ax1.add_artist(con)

Desenhar uma seta entre diferentes eixos

Vamos desenhar uma seta entre o mesmo ponto em coordenadas de dados, mas em diferentes eixos.

xy = (0.3, 0.2)
con = ConnectionPatch(
    xyA=xy, coordsA=ax2.transData,
    xyB=xy, coordsB=ax1.transData,
    arrowstyle="->", shrinkB=5)
fig.add_artist(con)

Desenhar uma linha entre diferentes pontos

Finalmente, vamos desenhar uma linha entre diferentes pontos, definidos em diferentes sistemas de coordenadas.

con = ConnectionPatch(
    ## in axes coordinates
    xyA=(0.6, 1.0), coordsA=ax2.transAxes,
    ## x in axes coordinates, y in data coordinates
    xyB=(0.0, 0.2), coordsB=ax2.get_yaxis_transform(),
    arrowstyle="-")
ax2.add_artist(con)

Resumo

Neste laboratório, aprendemos sobre o ConnectionPatch, que é usado para desenhar uma linha entre pontos definidos em diferentes sistemas de coordenadas e/ou eixos. Também aprendemos como desenhar uma seta simples e uma linha entre diferentes pontos em um gráfico.