barras_data_pico.py

# ----------------------------------------- Importando bibliotecas -----------------------------------------
import pandas as pd
import plotly.express as px
import tabela_utils

def faztudo(tabela : pd.DataFrame):
    tabela = tabela_utils.soma_por_categoria(tabela, 'MÊS', 'DECOLAGENS')
    tabela = tabela_utils.filtrar_colunas(tabela, ['DECOLAGENS'])
    tabela = (tabela['DECOLAGENS'].to_list())
    return tabela

def criar_grafico_barras_data_pico(Ano_selecionado, Mes_selecionado):
    # ---------------------------------------------- Lendo dataset ----------------------------------------------
    print('1 - Lendo dataset...') # Feedback
    dados = pd.read_csv('Dashboard-Oficial\data\ANAC20XX-13-14-15.csv', sep = ';', encoding = 'latin') # Encoding resolve problema da acentuação

    # --------------------------------------- Manipulando dados necessarios---------------------------------------
    print('1 - Filtrando os dados...') # Feedback
    filtrado = tabela_utils.filtrar_colunas(dados, ['ANO', 'MÊS', 'DECOLAGENS'])

    print('1 - Retirando os nulos...') # Feedback
    sem_nulo = tabela_utils.retirar_nulos(filtrado)

    print('1 - Separando anos...') # Feedback
    ano_1 = tabela_utils.filtrar_linhas(sem_nulo, 'ANO', '2013.0')
    ano_2 = tabela_utils.filtrar_linhas(sem_nulo, 'ANO', '2014.0')
    ano_3 = tabela_utils.filtrar_linhas(sem_nulo, 'ANO', '2015.0')

    print('1 - Fazendo mágica...') # Feedback
    ano_2013 = faztudo(ano_1)
    ano_2014 = faztudo(ano_2)
    ano_2015 = faztudo(ano_3)

    # --------------------------------------------- Criando Dataframe ---------------------------------------------
    ano_total = ano_2013 + ano_2014 + ano_2015
    meses = ["Janeiro", 'Fevereiro', 'Março', 'Abril', 'Maio', 'Junho', 'Julho', 'Agosto', 'Setembro', 'Outubro', 'Novembro', 'Dezembro']*3
    anos = ['2013']*12 + ['2014']*12 + ['2015']*12

    grafico = pd.DataFrame({
        "Mês": meses,
        "Decolagens": ano_total,
        "Ano": anos
    })

    print('1 - Filtrando os anos escolhidos...') # Feedback
    grafico = tabela_utils.filtrar_linhas(grafico, 'Ano', (Ano_selecionado))

    print('1 - Filtrando os meses escolhidos...') # Feedback
    graficofinal = tabela_utils.filtrar_linhas(grafico, 'Mês', (Mes_selecionado))


    # ----------------------------------------- Criando gráfico de barras -----------------------------------------
    print('1 - Produzindo gráfico...') # Feedback
    grafico_barras_data_pico = px.bar(graficofinal,
        x = "Mês", 
        y = "Decolagens", 
        color = "Ano", 
        barmode = "group", 
        color_discrete_sequence = px.colors.qualitative.Prism, 
        template = 'plotly_dark', 
        range_y = [80000, 100000]
        )

    return grafico_barras_data_pico