Como usar o diagrama de Sankey no Python?

[

Como criar um diagrama de Sankey em Python

Neste tutorial, vamos aprender como criar um diagrama de Sankey em Python. O diagrama de Sankey é uma visualização gráfica utilizada para representar fluxos direcionais de dados, como migração de populações, transferências de energia ou movimentos de mercadorias. É uma ferramenta eficaz para visualizar e analisar grandes conjuntos de dados com fluxos complexos.

Pré-requisitos

Antes de começarmos, certifique-se de ter a seguinte configuração em seu ambiente:

Python 3.x instalado
Biblioteca matplotlib instalada
Biblioteca numpy instalada

Passo 1: Importando as bibliotecas necessárias

Vamos iniciar o tutorial importando as bibliotecas necessárias para criar o diagrama de Sankey. Execute o seguinte código:

import matplotlib.pyplot as plt
import numpy as np

Passo 2: Criando os dados para o diagrama de Sankey

Agora, vamos criar alguns dados para o nosso diagrama de Sankey. Neste exemplo, iremos simular a migração populacional entre três regiões: A, B e C. Execute o seguinte código:

# Define as regiões
regioes = ['Região A', 'Região B', 'Região C']

# Define os fluxos e a proporção entre as regiões
fluxos = [100, 50, 20]
proporcao = [0.5, 0.4, 0.1]

# Cria uma matriz de fluxos
matriz_fluxos = np.column_stack((fluxos, proporcao))

# Cria os labels das regiões
label_regioes = ['{} - {}'.format(r, f) for r, f in zip(regioes, fluxos)]

Passo 3: Criando o diagrama de Sankey

Agora que temos os dados preparados, vamos criar o diagrama de Sankey. Execute o seguinte código:

# Cria uma figura e um subgráfico
fig, ax = plt.subplots(figsize=(10, 6))

# Plota o diagrama de Sankey
sankey = ax.sankey(flows=matriz_fluxos[:, 0],
                   labels=label_regioes,
                   label_position='right',
                   orientations=[0, 1, -1],
                   scale=0.1,
                   color='blue')

# Ajusta as configurações do gráfico
plt.axis('off')
plt.title('Diagrama de Sankey - Migração Populacional')

# Mostra o diagrama de Sankey
plt.show()

Conclusão

Neste tutorial, aprendemos como criar um diagrama de Sankey em Python. Vimos como importar as bibliotecas necessárias, criar os dados para o diagrama e plotar o diagrama de Sankey utilizando a biblioteca matplotlib. Agora, você pode usar essa técnica para visualizar e analisar fluxos de dados complexos em seus próprios projetos.

Experimente diferentes conjuntos de dados e personalize o diagrama de acordo com suas necessidades. Divirta-se explorando essa poderosa ferramenta de visualização em Python!