build_population_layouts: replace vresutils area calculation

This commit is contained in:
Fabian 2023-02-22 13:26:17 +01:00
parent 4b4990e432
commit dad7d2050e

View File

@ -7,8 +7,6 @@ import pandas as pd
import xarray as xr import xarray as xr
import geopandas as gpd import geopandas as gpd
from vresutils import shapes as vshapes
if __name__ == '__main__': if __name__ == '__main__':
if 'snakemake' not in globals(): if 'snakemake' not in globals():
from helper import mock_snakemake from helper import mock_snakemake
@ -46,8 +44,7 @@ if __name__ == '__main__':
pop_cells = pd.Series(I.dot(nuts3['pop'])) pop_cells = pd.Series(I.dot(nuts3['pop']))
# in km^2 # in km^2
with mp.Pool(processes=snakemake.threads) as pool: cell_areas = grid_cells.to_crs(3035).area / 1e6
cell_areas = pd.Series(pool.map(vshapes.area, grid_cells)) / 1e6
# pop per km^2 # pop per km^2
density_cells = pop_cells / cell_areas density_cells = pop_cells / cell_areas