taller-servel/helper.py

import contextily as ctx

def add_basemap(ax, zoom, url='http://tile.stamen.com/terrain/tileZ/tileX/tileY.png'):
    xmin, xmax, ymin, ymax = ax.axis()
    basemap, extent = ctx.bounds2img(xmin, ymin, xmax, ymax, zoom=zoom, url=url)
    ax.imshow(basemap, extent=extent, interpolation='bilinear')
    # restore original x/y limits
    ax.axis((xmin, xmax, ymin, ymax))


partidos_dict = {
    'PARTIDO DE TRABAJADORES REVOLUCIONARIOS': 'trabajadores revolucionarios',
    'DE TRABAJADORES REVOLUCIONARIOS': 'trabajadores revolucionarios',
    
    'UNION DE CENTRO CENTRO PROGRESISTA': 'uccp',
    'UNION DE CENTRO CENTRO PROGRES': 'uccp',
    
    'DE ACCION REGIONALISTA DE CHILE': 'ar',
    'DE ACCION REGIONALISTA DE CHIL': 'ar',
    
    'PARTIDO ECOLOGISTA VERDE': 'pev',
    'PARTIDO ECOLOGISTA VERDE ': 'pev',
    'ECOLOGISTA VERDE': 'pev',
    
    'P.A.I.S.': 'pa',
    'PAIS': 'pa',
    
    'DEMOCRACIA REGIONAL PATAGONICA': 'prp',
    'DEMOCRACIA REGIONAL PATAGONICA ': 'prp',
    
    'PARTIDO ECOLOGISTA VERDE DEL NORTE': 'pen',
    'ECOLOGISTA VERDE DEL NORTE': 'pen',
    
    'PARTIDO IZQUIERDA CIUDADANA DE CHILE': 'pic',
    'PARTIDO DE IZQUIERDA CIUDADANA DE CHILE': 'pic',
    'IZQUIERDA CIUDADANA DE CHILE': 'pic',
    
    'PARTIDO IGUALDAD': 'pi',
    'IGUALDAD': 'pi',
    
    'UNION DEMOCRATA INDEPENDIENTE': 'udi',
    'UNION DEMOCRATA INDEPENDIENTE ': 'udi',

    'PARTIDO POR LA DEMOCRACIA': 'ppd',
    'POR LA DEMOCRACIA ': 'ppd',
    'POR LA DEMOCRACIA': 'ppd',
    
    'PARTIDO EVOLUCION POLITICA ': 'pep',
    'EVOLUCION POLITICA': 'pep',
    
    'PARTIDO PROGRESISTA': 'pp',
    'PARTIDO PROGRESISTA ': 'pp',
    'PROGRESISTA': 'pp',
    
    'PARTIDO HUMANISTA': 'ph',
    'HUMANISTA': 'ph',
    
    'ALIANZA HUMANISTA VERDE': 'ah',
    'ALIANZA HUMANISTA-VERDE': 'ah',
    
    'DEMOCRATA CRISTIANO': 'pdc',
    'PARTIDO DEMOCRATA CRISTIANO': 'pdc',
    
    'RENOVACION NACIONAL': 'rn',
    'RENOVACION NACIONAL ': 'rn',
    
    'PARTIDO REGIONALISTA DE LOS INDEPENDIENTES': 'prdi',
    'PARTIDO REGIONALISTA INDEPENDIENTE': 'prdi',
    'REGIONALISTA DE LOS INDEPENDIE': 'prdi',
    'REGIONALISTA INDEPENDIENTE': 'pdri',
    'INDEPENDIENTE REGIONALISTA INDEPENDIENTE': 'pdri',

    'PARTIDO COMUNISTA DE CHILE': 'pc',
    'COMUNISTA DE CHILE': 'pc',
    'PARTIDO COMUNISTA DE CHILE ': 'pc',
    
    'PARTIDO LIBERAL DE CHILE': 'pl',
    'LIBERAL': 'pl',
    'LIBERAL DE CHILE': 'pl',
    
    'PARTIDO SOCIALISTA DE CHILE': 'ps',
    'SOCIALISTA CHILENO': 'ps',
    'SOCIALISTA DE CHILE': 'ps',
    'PARTIDO SOCIALISTA DE CHILE ': 'ps',
    
    'PARTIDO RADICAL SOCIALDEMOCRATA': 'prs',
    'RADICAL SOCIALDEMOCRATA': 'prs',
    'RADICAL SOCIAL DEMOCRATA': 'prs',
 
    'RADICAL DE CHILE': 'pr',
    'RADICAL': 'pr',
    
#     ' ': 'bl',
#     '  ': 'bl',
#     'nan': 'bl',
    
    'INDEPENDIENTE': 'independiente',
    'INDEPENDIENTES': 'independiente',
    'INDEPENDIENTES LISTA K': 'independiente',
    'INDEPENDIENTES LISTA A': 'independiente',
    'INDEPENDIENTES LISTA I': 'independiente',
    'INDEPENDIENTES LISTA G': 'independiente',
    'INDEPENDIENTES LISTA H': 'independiente',
    'INDEPENDIENTES LISTA J': 'independiente',
    'INDEPENDIENTES LISTA D': 'independiente',
    'INDEPENDIENTES LISTA B': 'independiente',
    'CANDIDATURA INDEPENDIENTE': 'independiente',
    'INDEPENDIENTE AMPLITUD': 'independiente',
    'INDEPENDIENTE PAIS': 'independiente',
    'INDEPENDIENTE REVOLUCION DEMOCRATICA': 'independiente',
    'INDEPENDIENTE LIBERAL DE CHILE': 'independiente',
    'INDEPENDIENTE CIUDADANOS': 'independiente',
    'INDEPENDIENTE UNION DEMOCRATA INDEPENDIENTE': 'independiente',
    'INDEPENDIENTE HUMANISTA': 'independiente',
    'INDEPENDIENTE DEMOCRACIA REGIONAL PATAGONICA': 'independiente',
    'INDEPENDIENTE RADICAL SOCIALDEMOCRATA': 'independiente',
    'INDEPENDIENTE SOCIALISTA DE CHILE': 'independiente',
    'INDEPENDIENTE MAS REGION': 'independiente',
    'INDEPENDIENTE POR LA DEMOCRACIA': 'independiente',
    'INDEPENDIENTE DEMOCRATA CRISTIANO': 'independiente',
    'INDEPENDIENTE RENOVACION NACIONAL': 'independiente',
    'INDEPENDIENTE PODER': 'independiente',
    'INDEPENDIENTE TODOS': 'independiente',
    'INDEPENDIENTE AMPLITUD': 'independiente',
    'INDEPENDIENTE PAIS': 'independiente',
    'INDEPENDIENTE REVOLUCION DEMOCRATICA': 'independiente',
    'INDEPENDIENTE LIBERAL DE CHILE': 'independiente',
    'INDEPENDIENTE CIUDADANOS': 'independiente',
    'INDEPENDIENTE UNION DEMOCRATA INDEPENDIENTE': 'independiente',
    'INDEPENDIENTE HUMANISTA': 'independiente',
    'INDEPENDIENTE DEMOCRACIA REGIONAL PATAGONICA': 'independiente',
    'INDEPENDIENTE RADICAL SOCIALDEMOCRATA': 'independiente',
    'INDEPENDIENTE SOCIALISTA DE CHILE': 'independiente',
    'INDEPENDIENTE MAS REGION': 'independiente',
    'INDEPENDIENTE COMUNISTA DE CHILE': 'independiente',
    'INDEPENDIENTE IZQUIERDA CIUDADANA DE CHILE': 'independiente',
    'INDEPENDIENTE EVOLUCION POLITICA': 'independiente',
    'INDEPENDIENTE FEDERACION REGIONALISTA VERDE SOCIAL': 'independiente',
    'INDEPENDIENTE ECOLOGISTA VERDE': 'independiente',
    'INDEPENDIENTE PROGRESISTA': 'independiente',
    'ALIANZA NACIONAL DE LOS INDEPE': 'independiente',
    'INDEPENDIENTE IGUALDAD': 'independiente',
}

columns_to_drop = [
    'Fecha de Elección',
    'Partido',
    'Inicio de Período',
    'Fin de Período',
    'Id Región'
]

comunas_servel2bnc = {
    'ARICA': 'Arica',
    'CAMARONES': 'Camarones',
    'GENERAL LAGOS': 'General Lagos',
    'PUTRE': 'Putre',
    'ALTO HOSPICIO': 'Alto Hospicio',
    'CAMIÑA': 'Camiña',
    'COLCHANE': 'Colchane',
    'HUARA': 'Huara',
    'IQUIQUE': 'Iquique',
    'PICA': 'Pica',
    'POZO ALMONTE': 'Pozo Almonte',
    'CALAMA': 'Calama',
    'MARIA ELENA': 'María Elena',
    'OLLAGUE': 'Ollagüe',
    'SAN PEDRO DE ATACAMA': 'San Pedro de Atacama',
    'TOCOPILLA': 'Tocopilla',
    'ANTOFAGASTA': 'Antofagasta',
    'MEJILLONES': 'Mejillones',
    'SIERRA GORDA': 'Sierra Gorda',
    'TALTAL': 'Taltal',
    'CHAÑARAL': 'Chañaral',
    'COPIAPO': 'Copiapó',
    'DIEGO DE ALMAGRO': 'Diego de Almagro',
    'ALTO DEL CARMEN': 'Alto del Carmen',
    'CALDERA': 'Caldera',
    'FREIRINA': 'Freirina',
    'HUASCO': 'Huasco',
    'TIERRA AMARILLA': 'Tierra Amarilla',
    'VALLENAR': 'Vallenar',
    'ANDACOLLO': 'Andacollo',
    'LA HIGUERA': 'La Higuera',
    'LA SERENA': 'La Serena',
    'PAIHUANO': 'Paihuano',
    'VICUÑA': 'Vicuña',
    'COQUIMBO': 'Coquimbo',
    'OVALLE': 'Ovalle',
    'RIO HURTADO': 'Río Hurtado',
    'CANELA': 'Canela',
    'COMBARBALA': 'Combarbalá',
    'ILLAPEL': 'Illapel',
    'LOS VILOS': 'Los Vilos',
    'MONTE PATRIA': 'Monte Patria',
    'PUNITAQUI': 'Punitaqui',
    'SALAMANCA': 'Salamanca',
    'CABILDO': 'Cabildo',
    'CALERA': 'Calera',
    'HIJUELAS': 'Hijuelas',
    'LA CRUZ': 'La Cruz',
    'LA LIGUA': 'La Ligua',
    'NOGALES': 'Nogales',
    'PAPUDO': 'Papudo',
    'PETORCA': 'Petorca',
    'PUCHUNCAVI': 'Puchuncaví',
    'QUILLOTA': 'Quillota',
    'QUINTERO': 'Quintero',
    'ZAPALLAR': 'Zapallar',
    'CALLE LARGA': 'Calle Larga',
    'CATEMU': 'Catemu',
    'LOS ANDES': 'Los Andes',
    'PANQUEHUE': 'Panquehue',
    'PUTAENDO': 'Putaendo',
    'RINCONADA': 'Rinconada',
    'SAN ESTEBAN': 'San Esteban',
    'SAN FELIPE': 'San Felipe',
    'SANTA MARIA': 'Santa María',
    'LIMACHE': 'Limache',
    'OLMUE': 'Olmué',
    'QUILPUE': 'Quilpué',
    'VILLA ALEMANA': 'Villa Alemana',
    'ISLA DE PASCUA': 'Isla de Pascua',
    'JUAN FERNANDEZ': 'Juan Fernández',
    'VALPARAISO': 'Valparaíso',
    'CONCON': 'Concón',
    'VIÑA DEL MAR': 'Viña del Mar',
    'ALGARROBO': 'Algarrobo',
    'CARTAGENA': 'Cartagena',
    'CASABLANCA': 'Casablanca',
    'EL QUISCO': 'El Quisco',
    'EL TABO': 'El Tabo',
    'SAN ANTONIO': 'San Antonio',
    'SANTO DOMINGO': 'Santo Domingo',
    'COLINA': 'Colina',
    'LAMPA': 'Lampa',
    'PUDAHUEL': 'Pudahuel',
    'QUILICURA': 'Quilicura',
    'TILTIL': 'Tiltil',
    'CONCHALI': 'Conchalí',
    'HUECHURABA': 'Huechuraba',
    'RENCA': 'Renca',
    'CERRO NAVIA': 'Cerro Navia',
    'LO PRADO': 'Lo Prado',
    'QUINTA NORMAL': 'Quinta Normal',
    'INDEPENDENCIA': 'Independencia',
    'RECOLETA': 'Recoleta',
    'CERRILLOS': 'Cerrillos',
    'ESTACION CENTRAL': 'Estación Central',
    'MAIPU': 'Maipú',
    'ÑUÑOA': 'Ñuñoa',
    'PROVIDENCIA': 'Providencia',
    'SANTIAGO': 'Santiago',
    'LAS CONDES': 'Las Condes',
    'LO BARNECHEA': 'Lo Barnechea',
    'VITACURA': 'Vitacura',
    'LA REINA': 'La Reina',
    'PEÑALOLEN': 'Peñalolén',
    'LA GRANJA': 'La Granja',
    'MACUL': 'Macul',
    'SAN JOAQUIN': 'San Joaquín',
    'LA FLORIDA': 'La Florida',
    'EL BOSQUE': 'El Bosque',
    'LA CISTERNA': 'La Cisterna',
    'SAN RAMON': 'San Ramón',
    'LO ESPEJO': 'Lo Espejo',
    'PEDRO AGUIRRE CERDA': 'Pedro Aguirre Cerda',
    'SAN MIGUEL': 'San Miguel',
    'LA PINTANA': 'La Pintana',
    'PIRQUE': 'Pirque',
    'PUENTE ALTO': 'Puente Alto',
    'SAN JOSE DE MAIPO': 'San José de Maipo',
    'BUIN': 'Buin',
    'CALERA DE TANGO': 'Calera de Tango',
    'PAINE': 'Paine',
    'SAN BERNARDO': 'San Bernardo',
    'ALHUE': 'Alhué',
    'CURACAVI': 'Curacaví',
    'EL MONTE': 'El Monte',
    'ISLA DE MAIPO': 'Isla de Maipo',
    'MARIA PINTO': 'María Pinto',
    'MELIPILLA': 'Melipilla',
    'PADRE HURTADO': 'Padre Hurtado',
    'PEÑAFLOR': 'Peñaflor',
    'SAN PEDRO': 'San Pedro',
    'TALAGANTE': 'Talagante',
    'RANCAGUA': 'Rancagua',
    'CODEGUA': 'Codegua',
    'COINCO': 'Coinco',
    'COLTAUCO': 'Coltauco',
    'DOÑIHUE': 'Doñihue',
    'GRANEROS': 'Graneros',
    'MACHALI': 'Machalí',
    'MALLOA': 'Malloa',
    'MOSTAZAL': 'Mostazal',
    'OLIVAR': 'Olivar',
    'QUINTA DE TILCOCO': 'Quinta de Tilcoco',
    'RENGO': 'Rengo',
    'REQUINOA': 'Requínoa',
    'CHIMBARONGO': 'Chimbarongo',
    'LAS CABRAS': 'Las Cabras',
    'PEUMO': 'Peumo',
    'PICHIDEGUA': 'Pichidegua',
    'SAN FERNANDO': 'San Fernando',
    'SAN VICENTE': 'San Vicente',
    'CHEPICA': 'Chépica',
    'LA ESTRELLA': 'La Estrella',
    'LITUECHE': 'Litueche',
    'LOLOL': 'Lolol',
    'MARCHIGUE': 'Marchigüe',
    'NANCAGUA': 'Nancagua',
    'NAVIDAD': 'Navidad',
    'PALMILLA': 'Palmilla',
    'PAREDONES': 'Paredones',
    'PERALILLO': 'Peralillo',
    'PICHILEMU': 'Pichilemu',
    'PLACILLA': 'Placilla',
    'PUMANQUE': 'Pumanque',
    'SANTA CRUZ': 'Santa Cruz',
    'CURICO': 'Curicó',
    'HUALAÑE': 'Hualañé',
    'LICANTEN': 'Licantén',
    'MOLINA': 'Molina',
    'RAUCO': 'Rauco',
    'ROMERAL': 'Romeral',
    'SAGRADA FAMILIA': 'Sagrada Familia',
    'TENO': 'Teno',
    'VICHUQUEN': 'Vichuquén',
    'TALCA': 'Talca',
    'CONSTITUCION': 'Constitución',
    'CUREPTO': 'Curepto',
    'EMPEDRADO': 'Empedrado',
    'MAULE': 'Maule',
    'PELARCO': 'Pelarco',
    'PENCAHUE': 'Pencahue',
    'RIO CLARO': 'Río Claro',
    'SAN CLEMENTE': 'San Clemente',
    'SAN RAFAEL': 'San Rafael',
    'COLBUN': 'Colbún',
    'LINARES': 'Linares',
    'SAN JAVIER': 'San Javier',
    'VILLA ALEGRE': 'Villa Alegre',
    'YERBAS BUENAS': 'Yerbas Buenas',
    'CAUQUENES': 'Cauquenes',
    'CHANCO': 'Chanco',
    'LONGAVI': 'Longaví',
    'PARRAL': 'Parral',
    'PELLUHUE': 'Pelluhue',
    'RETIRO': 'Retiro',
    'CHILLAN': 'Chillán',
    'CHILLAN VIEJO': 'Chillán Viejo',
    'COIHUECO': 'Coihueco',
    'EL CARMEN': 'El Carmen',
    'PEMUCO': 'Pemuco',
    'PINTO': 'Pinto',
    'SAN IGNACIO': 'San Ignacio',
    'YUNGAY': 'Yungay',
    'BULNES': 'Bulnes',
    'CABRERO': 'Cabrero',
    'COBQUECURA': 'Cobquecura',
    'COELEMU': 'Coelemu',
    'NINHUE': 'Ninhue',
    'ÑIQUEN': 'Ñiquén',
    'PORTEZUELO': 'Portezuelo',
    'QUILLON': 'Quillón',
    'QUIRIHUE': 'Quirihue',
    'RANQUIL': 'Ranquil',
    'SAN CARLOS': 'San Carlos',
    'SAN FABIAN': 'San Fabián',
    'SAN NICOLAS': 'San Nicolás',
    'TREGUACO': 'Treguaco',
    'YUMBEL': 'Yumbel',
    'HUALPEN': 'Hualpén',
    'TALCAHUANO': 'Talcahuano',
    'CHIGUAYANTE': 'Chiguayante',
    'CONCEPCION': 'Concepción',
    'SAN PEDRO DE LA PAZ': 'San Pedro de la Paz',
    'CORONEL': 'Coronel',
    'FLORIDA': 'Florida',
    'HUALQUI': 'Hualqui',
    'PENCO': 'Penco',
    'SANTA JUANA': 'Santa Juana',
    'TOME': 'Tomé',
    'ARAUCO': 'Arauco',
    'CAÑETE': 'Cañete',
    'CONTULMO': 'Contulmo',
    'CURANILAHUE': 'Curanilahue',
    'LEBU': 'Lebu',
    'LOS ALAMOS': 'Los Alamos',
    'LOTA': 'Lota',
    'TIRUA': 'Tirúa',
    'ALTO BIOBIO': 'Alto Biobío',
    'ANTUCO': 'Antuco',
    'LAJA': 'Laja',
    'LOS ANGELES': 'Los Angeles',
    'MULCHEN': 'Mulchén',
    'NACIMIENTO': 'Nacimiento',
    'NEGRETE': 'Negrete',
    'QUILACO': 'Quilaco',
    'QUILLECO': 'Quilleco',
    'SAN ROSENDO': 'San Rosendo',
    'SANTA BARBARA': 'Santa Bárbara',
    'TUCAPEL': 'Tucapel',
    'ANGOL': 'Angol',
    'COLLIPULLI': 'Collipulli',
    'ERCILLA': 'Ercilla',
    'LOS SAUCES': 'Los Sauces',
    'LUMACO': 'Lumaco',
    'PUREN': 'Purén',
    'RENAICO': 'Renaico',
    'TRAIGUEN': 'Traiguén',
    'CURACAUTIN': 'Curacautín',
    'GALVARINO': 'Galvarino',
    'LAUTARO': 'Lautaro',
    'LONQUIMAY': 'Lonquimay',
    'MELIPEUCO': 'Melipeuco',
    'PERQUENCO': 'Perquenco',
    'VICTORIA': 'Victoria',
    'VILCUN': 'Vilcún',
    'PADRE LAS CASAS': 'Padre Las Casas',
    'TEMUCO': 'Temuco',
    'CARAHUE': 'Carahue',
    'CHOLCHOL': 'Cholchol',
    'FREIRE': 'Freire',
    'NUEVA IMPERIAL': 'Nueva Imperial',
    'PITRUFQUEN': 'Pitrufquén',
    'SAAVEDRA': 'Saavedra',
    'TEODORO SCHMIDT': 'Teodoro Schmidt',
    'CUNCO': 'Cunco',
    'CURARREHUE': 'Curarrehue',
    'GORBEA': 'Gorbea',
    'LONCOCHE': 'Loncoche',
    'PUCON': 'Pucón',
    'TOLTEN': 'Toltén',
    'VILLARRICA': 'Villarrica',
    'CORRAL': 'Corral',
    'LANCO': 'Lanco',
    'MAFIL': 'Máfil',
    'MARIQUINA': 'Mariquina',
    'VALDIVIA': 'Valdivia',
    'FUTRONO': 'Futrono',
    'LA UNION': 'La Unión',
    'LAGO RANCO': 'Lago Ranco',
    'LOS LAGOS': 'Los Lagos',
    'PAILLACO': 'Paillaco',
    'PANGUIPULLI': 'Panguipulli',
    'RIO BUENO': 'Río Bueno',
    'OSORNO': 'Osorno',
    'SAN JUAN DE LA COSTA': 'San Juan de la Costa',
    'SAN PABLO': 'San Pablo',
    'FRESIA': 'Fresia',
    'FRUTILLAR': 'Frutillar',
    'LLANQUIHUE': 'Llanquihue',
    'LOS MUERMOS': 'Los Muermos',
    'PUERTO OCTAY': 'Puerto Octay',
    'PUERTO VARAS': 'Puerto Varas',
    'PURRANQUE': 'Purranque',
    'PUYEHUE': 'Puyehue',
    'RIO NEGRO': 'Río Negro',
    'CALBUCO': 'Calbuco',
    'COCHAMO': 'Cochamó',
    'MAULLIN': 'Maullín',
    'PUERTO MONTT': 'Puerto Montt',
    'ANCUD': 'Ancud',
    'CASTRO': 'Castro',
    'CHAITEN': 'Chaitén',
    'CHONCHI': 'Chonchi',
    'CURACO DE VELEZ': 'Curaco de Vélez',
    'DALCAHUE': 'Dalcahue',
    'FUTALEUFU': 'Futaleufú',
    'HUALAIHUE': 'Hualaihué',
    'PALENA': 'Palena',
    'PUQUELDON': 'Puqueldón',
    'QUEILEN': 'Queilén',
    'QUELLON': 'Quellón',
    'QUEMCHI': 'Quemchi',
    'QUINCHAO': 'Quinchao',
    'CHILE CHICO': 'Chile Chico',
    'CISNES': 'Cisnes',
    'COCHRANE': 'Cochrane',
    'COYHAIQUE': 'Coyhaique',
    'GUAITECAS': 'Guaitecas',
    'LAGO VERDE': 'Lago Verde',
    'O\'HIGGINS': 'O\'Higgins',
    'RIO IBAÑEZ': 'Río Ibáñez',
    'TORTEL': 'Tortel',
    'CABO DE HORNOS': 'Cabo de Hornos',
    'LAGUNA BLANCA': 'Laguna Blanca',
    'NATALES': 'Natales',
    'PORVENIR': 'Porvenir',
    'PRIMAVERA': 'Primavera',
    'PUNTA ARENAS': 'Punta Arenas',
    'RIO VERDE': 'Río Verde',
    'SAN GREGORIO': 'San Gregorio',
    'TIMAUKEL': 'Timaukel',
    'TORRES DEL PAINE': 'Torres del Paine',
    'AISEN': 'Aysen',
    'LLAY-LLAY': 'Llaillay',
    'ANTARTICA': 'Cabo de Hornos',
    'CABO DE HORNOS Y ANTARTICA': 'Cabo de Hornos',
    'NAVARINO': 'Cabo de Hornos'
}

def calcula_partido(partido_string):
    if type(partido_string) is not str:
        return partido_string
    if partido_string.strip() in partidos_dict:
        return partidos_dict[partido_string.strip()]
    else:
        return partido_string
    
    
def clear_electo(electo_str):
    if type(electo_str) != str:
        return electo_str
    elif len(electo_str.strip()) > 0:
        return electo_str.strip()
    else:
        return None
    
    
def set_partido(row):
    if row['Candidato (a)'][:11] == 'VOTOS NULOS':
        row['Partido'] = 'Nulos'
    elif row['Candidato (a)'][:15] == 'VOTOS EN BLANCO':
        row['Partido'] = 'Blancos'
        
    return row

    
def clear_df(df):
    df = df.apply(set_partido, axis=1)
    df['partido'] = df.Partido.apply(calcula_partido)
    df['electo'] = df['Electo(a)'].apply(clear_electo)
    df['comuna'] = df.Comuna.apply(lambda x: comunas_servel2bnc[x])
    df['nombre'] = df.Nombres.str.strip() + ' ' + df['Apellido Paterno'].str.strip() + ' ' + df['Apellido Materno'].str.strip()
    df['id_elección'] = df['Tipo de Elección'].str.lower() + ' ' + df['Fecha de Elección'].astype(str).str[:4]
    
    return df.drop(columns=columns_to_drop)