diff --git a/classificacao_veranicos.py b/classificacao_veranicos.py index 4ebfb31..8a02be4 100644 --- a/classificacao_veranicos.py +++ b/classificacao_veranicos.py @@ -76,13 +76,9 @@ def rodar_decendio(decendios,qtd_veranicos, estacoes): filtro = decendios[(decendios['INICIO'] <= data_fim) & (decendios['FINAL'] >= data_inicio)] dia =selecao_estacao['QTDDIAS'] - # Adicionar colunas FINICIO e FFINAL - filtro = filtro.copy() # Para evitar avisos do Pandas - #filtro['FINICIO'] = data_inicio - #filtro['FFINAL'] = data_fim - # filtro['dias'] = filtro.apply(lambda row: diferenca_dias(row['INICIO'], row['FINAL']), axis=1) + filtro = filtro.copy() # Para evitar avisos do Pandas filtro['dias_chuva'] = 0 - #filtro.apply(lambda row: diferenca_dias(row['FINICIO'], row['FINAL']), axis=1) + filtro['dias_decendio'] = 0 @@ -97,22 +93,29 @@ def rodar_decendio(decendios,qtd_veranicos, estacoes): elif quantidade_veranicos <= 1: print(f"Tem mais de 1 registros ({quantidade_veranicos} linhas).") filtro.loc[filtro.index[0], 'dias_chuva'] = diferenca_dias(data_inicio, data_fim)+1 + filtro.loc[filtro.index[0], 'dias_decendio'] =diferenca_dias(filtro.iloc[0]['INICIO'], filtro.iloc[0]['FINAL'])+1 - print(filtro) + elif quantidade_veranicos <= 2: print(f"Tem mais de 2 registros ({quantidade_veranicos} linhas).") filtro.loc[filtro.index[0], 'dias_chuva'] = diferenca_dias(data_inicio, filtro.iloc[0]['FINAL'])+1 filtro.loc[filtro.index[1], 'dias_chuva'] = diferenca_dias(filtro.iloc[1]['INICIO'], data_fim)+1 + filtro.loc[filtro.index[0], 'dias_decendio'] =diferenca_dias(filtro.iloc[0]['INICIO'], filtro.iloc[0]['FINAL'])+1 + filtro.loc[filtro.index[1], 'dias_decendio'] =diferenca_dias(filtro.iloc[1]['INICIO'], filtro.iloc[1]['FINAL'])+1 + - print(filtro) + elif quantidade_veranicos <= 3: print(f"Tem mais de 3 registros ({quantidade_veranicos} linhas).") filtro.loc[filtro.index[0], 'dias_chuva'] = diferenca_dias(data_inicio, filtro.iloc[0]['FINAL'])+1 filtro.loc[filtro.index[1], 'dias_chuva'] = diferenca_dias(filtro.iloc[1]['INICIO'], filtro.iloc[1]['FINAL']) +1 filtro.loc[filtro.index[2], 'dias_chuva'] = diferenca_dias(filtro.iloc[2]['INICIO'], data_fim)+1 - print(filtro) + filtro.loc[filtro.index[0], 'dias_decendio'] =diferenca_dias(filtro.iloc[0]['INICIO'], filtro.iloc[0]['FINAL'])+1 + filtro.loc[filtro.index[1], 'dias_decendio'] =diferenca_dias(filtro.iloc[1]['INICIO'], filtro.iloc[1]['FINAL'])+1 + filtro.loc[filtro.index[2], 'dias_decendio'] =diferenca_dias(filtro.iloc[2]['INICIO'], filtro.iloc[2]['FINAL'])+1 + elif quantidade_veranicos <= 4: print(f"Tem mais de 4 registros ({quantidade_veranicos} linhas).") @@ -120,9 +123,15 @@ def rodar_decendio(decendios,qtd_veranicos, estacoes): filtro.loc[filtro.index[1], 'dias_chuva'] = diferenca_dias(filtro.iloc[1]['INICIO'], filtro.iloc[1]['FINAL']) filtro.loc[filtro.index[2], 'dias_chuva'] = diferenca_dias(filtro.iloc[2]['INICIO'], filtro.iloc[2]['FINAL']) filtro.loc[filtro.index[3], 'dias_chuva'] = diferenca_dias(filtro.iloc[3]['INICIO'], data_fim)+1 - print(filtro) + + filtro.loc[filtro.index[0], 'dias_decendio'] =diferenca_dias(filtro.iloc[0]['INICIO'], filtro.iloc[0]['FINAL'])+1 + filtro.loc[filtro.index[1], 'dias_decendio'] =diferenca_dias(filtro.iloc[1]['INICIO'], filtro.iloc[1]['FINAL'])+1 + filtro.loc[filtro.index[2], 'dias_decendio'] =diferenca_dias(filtro.iloc[2]['INICIO'], filtro.iloc[2]['FINAL'])+1 + filtro.loc[filtro.index[3], 'dias_decendio'] =diferenca_dias(filtro.iloc[3]['INICIO'], filtro.iloc[3]['FINAL'])+1 + + print(filtro) dados = [] - print('->', selecao_estacao) + # print('->', selecao_estacao) for index, row in filtro.iterrows(): dados.append({ 'CODIGO': selecao_estacao['CODIGO'],