From 6b29f34d5181bd4711600b1ef16164b5a290058a Mon Sep 17 00:00:00 2001 From: Jonas Hoersch Date: Wed, 19 Dec 2018 15:42:03 +0100 Subject: [PATCH] build_shapes: Ensure name of index for offshore_shapes is 'name' --- scripts/build_shapes.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/build_shapes.py b/scripts/build_shapes.py index 54e5fcba..b8dba2e0 100644 --- a/scripts/build_shapes.py +++ b/scripts/build_shapes.py @@ -52,7 +52,9 @@ def eez(country_shapes): df = df.loc[df['ISO_3digit'].isin([_get_country('alpha_3', alpha_2=c) for c in snakemake.config['countries']])] df['name'] = df['ISO_3digit'].map(lambda c: _get_country('alpha_2', alpha_3=c)) s = df.set_index('name').geometry.map(lambda s: _simplify_polys(s, filterremote=False)) - return gpd.GeoSeries({k:v for k,v in s.iteritems() if v.distance(country_shapes[k]) < 1e-3}) + s = gpd.GeoSeries({k:v for k,v in s.iteritems() if v.distance(country_shapes[k]) < 1e-3}) + s.index.name = "name" + return s def country_cover(country_shapes, eez_shapes=None): shapes = list(country_shapes)