calculo_poblacion.py

from shapely.geometry import Point, Polygon, mapping
import matplotlib.pyplot as plt
import glob
import os
import pandas as pd
import geopandas as gpd
from shapely import wkt
import openpyxl
import os


def run(args,dataset_filename,shp_filename,report_filename):
    #Importamos los puntos de poblacion
    inegi_path='./datos/datasets/INEGI_2010/Localidades_20100_semicolon.csv'
    inegi_data = pd.read_csv(inegi_path,header=0, delimiter = ";", encoding = 'latin-1')
    #Importamos los shp de cobertura
    coberturas_dir = args.folder #Local path, you can change it to mkae it matc your files route. i.e. '/csv/' , '../../Documents/"
    coberturas_dir_list = glob.glob(os.path.join(coberturas_dir, '*.shp'))
    print("Archivos de cobertura incluidos: ", len(coberturas_dir_list))

    # datos para almacenar resultados

    gdf_base = gpd.GeoDataFrame(
        inegi_data, geometry=gpd.points_from_xy(inegi_data.LONGITUD, inegi_data.LATITUD))

    gdf_base.set_crs(epsg=4326, inplace=True)

    localidades_incluidas = set()

    for cobertura in coberturas_dir_list:
        cobertura_shp= gpd.read_file(cobertura)
        points_within = gpd.sjoin(gdf_base,cobertura_shp, predicate='within')
        print("Procesada estacion: ", cobertura)
        
        localidades_incluidas.update(points_within["_ID"].to_list())
    
    if (len(localidades_incluidas))<1: 
        print("No hay archivos en la ruta especificada, terminando ejecución")
    else:
        # print(localidades_incluidas)
        print(len(localidades_incluidas), " Localidades incluidas en el grupo ", args.nombre_destino)
        localidades_incluidas_df = pd.DataFrame(list(localidades_incluidas),columns = ['_ID']) 

        result_coverage = pd.merge(localidades_incluidas_df,inegi_data, on='_ID', how='left')
        # print(result_coverage)

        writer = pd.ExcelWriter((dataset_filename),engine='openpyxl')
        result_coverage.to_excel(writer,index = True, header=True)
        writer.save()

        # Geopandas GeoDataFrame
        result_coverage_shp = gpd.GeoDataFrame(result_coverage, geometry='geometry')
        #Export to shapefile
        result_coverage_shp.to_file(shp_filename)

        extension_territorial =  result_coverage['SUPERFICIE'].sum(axis=0)
        print("Extension territorial de las localidades incluidas: ", extension_territorial, " km^2")

        results = {'Nombre del grupo: ': args.nombre_destino,'Numero de localidades: ': len(coberturas_dir_list),'Extensión territorial: ':extension_territorial}
        ##Guardamos resultados
        def write_results():
            with open(report_filename, 'w') as f:
                for key in results.keys():
                    f.write("%s, %s\n" % (key, results[key]))
                    
        write_results()
        print ('Listo')