Resultados Eleitorais

https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_candidato_munzona/votacao_candidato_munzona_2022.zip
https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_secao/votacao_secao_2022_BR.zip

https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_candidato_munzona/votacao_candidato_munzona_2022.zip
https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_candidato_munzona/votacao_candidato_munzona_2020.zip
https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_candidato_munzona/votacao_candidato_munzona_2018.zip

https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_secao/votacao_secao_2022_BR.zip
https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_secao/votacao_secao_2022_SP.zip
https://cdn.tse.jus.br/estatistica/sead/odsele/votacao_secao/votacao_secao_2022_RJ.zip

import requests
import zipfile
import pandas as pd
from io import BytesIO
from pathlib import Path
 
 
def baixar_votacao_munzona(ano: int, destino: Path = Path("./dados_tse")) -> pd.DataFrame:
    """
    Baixa e carrega dados de votação por candidato/município/zona.
 
    Args:
        ano: Ano da eleição (ex: 2022, 2020, 2018)
        destino: Diretório para salvar os arquivos
 
    Returns:
        DataFrame com votação por candidato e município/zona
    """
    url = (
        f"https://cdn.tse.jus.br/estatistica/sead/odsele/"
        f"votacao_candidato_munzona/votacao_candidato_munzona_{ano}.zip"
    )
    print(f"Baixando votação {ano}...")
 
    response = requests.get(url)
    response.raise_for_status()
 
    destino.mkdir(parents=True, exist_ok=True)
 
    with zipfile.ZipFile(BytesIO(response.content)) as zf:
        arquivo_brasil = [
            f for f in zf.namelist()
            if "BRASIL" in f.upper() and f.endswith(".csv")
        ]
        if arquivo_brasil:
            zf.extract(arquivo_brasil[0], destino)
            caminho_csv = destino / arquivo_brasil[0]
        else:
            zf.extractall(destino)
            csvs = [f for f in zf.namelist() if f.endswith(".csv")]
            caminho_csv = destino / csvs[0]
 
    df = pd.read_csv(
        caminho_csv,
        sep=";",
        encoding="latin-1",
        dtype=str,
    )
 
    print(f"Registros carregados: {len(df):,}")
    return df
 
 
# Carregar resultados das eleições de 2022
df_votos = baixar_votacao_munzona(2022)
print(df_votos.head())

def votacao_por_municipio(
    df: pd.DataFrame,
    numero_candidato: str,
    cargo: str = None,
    turno: str = "1",
) -> pd.DataFrame:
    """
    Agrega a votação de um candidato por município.
 
    Args:
        df: DataFrame com dados de votação
        numero_candidato: Número do candidato na urna
        cargo: Código ou descrição do cargo (opcional)
        turno: Turno da eleição ('1' ou '2')
 
    Returns:
        DataFrame com votação por município, ordenado por votos
    """
    filtro = (
        (df["NR_CANDIDATO"] == str(numero_candidato))
        & (df["NR_TURNO"] == str(turno))
    )
    if cargo:
        filtro = filtro & (df["DS_CARGO"].str.upper() == cargo.upper())
 
    df_filtrado = df[filtro].copy()
    df_filtrado["QT_VOTOS_NOMINAIS"] = pd.to_numeric(
        df_filtrado["QT_VOTOS_NOMINAIS"], errors="coerce"
    )
 
    resultado = (
        df_filtrado.groupby(["CD_MUNICIPIO", "NM_MUNICIPIO", "SG_UF"])
        .agg(total_votos=("QT_VOTOS_NOMINAIS", "sum"))
        .reset_index()
        .sort_values("total_votos", ascending=False)
    )
 
    return resultado
 
 
# Exemplo: votação por município de um candidato a presidente
resultado = votacao_por_municipio(df_votos, "13", cargo="PRESIDENTE")
print(f"Municípios com votação: {len(resultado)}")
print(resultado.head(20))

def carregar_votacao_secao(ano: int, uf: str) -> pd.DataFrame:
    """
    Carrega votação por seção eleitoral de um estado.
 
    Args:
        ano: Ano da eleição
        uf: Sigla da UF (ex: 'SP', 'RJ')
 
    Returns:
        DataFrame com votação por seção eleitoral
    """
    url = (
        f"https://cdn.tse.jus.br/estatistica/sead/odsele/"
        f"votacao_secao/votacao_secao_{ano}_{uf}.zip"
    )
    response = requests.get(url)
    response.raise_for_status()
 
    with zipfile.ZipFile(BytesIO(response.content)) as zf:
        csvs = [f for f in zf.namelist() if f.endswith(".csv")]
        zf.extractall(Path("./dados_tse"))
 
    df = pd.read_csv(
        Path("./dados_tse") / csvs[0],
        sep=";",
        encoding="latin-1",
        dtype=str,
    )
    return df
 
 
# Carregar votação por seção em São Paulo
df_secao_sp = carregar_votacao_secao(2022, "SP")
print(f"Registros por seção (SP): {len(df_secao_sp):,}")

import pandas as pd
 
# 1. Carregar resultados eleitorais
df_votos = pd.read_csv(
    "dados_tse/votacao_candidato_munzona_2022_BRASIL.csv",
    sep=";",
    encoding="latin-1",
    dtype=str,
)
 
# 2. Agregar votos totais por candidato
df_votos["QT_VOTOS_NOMINAIS"] = pd.to_numeric(
    df_votos["QT_VOTOS_NOMINAIS"], errors="coerce"
)
votos_totais = (
    df_votos.groupby(["SQ_CANDIDATO", "NM_CANDIDATO", "SG_PARTIDO", "DS_CARGO"])
    .agg(total_votos=("QT_VOTOS_NOMINAIS", "sum"))
    .reset_index()
)
 
# 3. Carregar prestação de contas (receitas totais por candidato)
df_receitas = pd.read_csv(
    "dados_tse/prestacao_contas_2022_receitas_BRASIL.csv",
    sep=";",
    encoding="latin-1",
    dtype=str,
)
df_receitas["VR_RECEITA"] = pd.to_numeric(
    df_receitas["VR_RECEITA"].str.replace(",", "."), errors="coerce"
)
receitas_totais = (
    df_receitas.groupby("SQ_CANDIDATO")
    .agg(total_receitas=("VR_RECEITA", "sum"))
    .reset_index()
)
 
# 4. Cruzar votos com receitas
analise = pd.merge(votos_totais, receitas_totais, on="SQ_CANDIDATO", how="inner")
analise["custo_por_voto"] = analise["total_receitas"] / analise["total_votos"]
 
print("Top 10 candidatos — custo por voto (deputados federais):")
dep_fed = analise[analise["DS_CARGO"] == "DEPUTADO FEDERAL"].sort_values(
    "custo_por_voto", ascending=True
)
print(dep_fed[["NM_CANDIDATO", "SG_PARTIDO", "total_votos", "total_receitas", "custo_por_voto"]].head(10))