bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2023-08-21 14:17:19 +00:00
parent 8d09e38627
commit dadc372ecd
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/build_industrial_distribution_key.py
View File

@ -96,12 +96,15 @@ def prepare_hotmaps_database(regions):
# the .sjoin can lead to duplicates if a geom is in two regions # the .sjoin can lead to duplicates if a geom is in two regions
if gdf.index.duplicated().any(): if gdf.index.duplicated().any():
import pycountry import pycountry
# get all duplicated entries # get all duplicated entries
duplicated_i = gdf.index[gdf.index.duplicated()] duplicated_i = gdf.index[gdf.index.duplicated()]
# convert from raw data country name to iso-2-code # convert from raw data country name to iso-2-code
s = df.loc[duplicated_i, "Country"].apply(lambda x: pycountry.countries.lookup(x).alpha_2) s = df.loc[duplicated_i, "Country"].apply(
lambda x: pycountry.countries.lookup(x).alpha_2
)
# Get a boolean mask where gdf's country column matches s's values for the same index # Get a boolean mask where gdf's country column matches s's values for the same index
mask = gdf['country'] == gdf.index.map(s) mask = gdf["country"] == gdf.index.map(s)
# Filter gdf using the mask # Filter gdf using the mask
gdf_filtered = gdf[mask] gdf_filtered = gdf[mask]
# concat not duplicated and filtered gdf # concat not duplicated and filtered gdf