diff --git a/swmmio/utils/spatial.py b/swmmio/utils/spatial.py index 8dd7812..e3570c6 100644 --- a/swmmio/utils/spatial.py +++ b/swmmio/utils/spatial.py @@ -216,6 +216,8 @@ def write_shapefile(df, filename, geomtype='line', prj=None): w.line([row.coords]) if geomtype == 'point': w.point(*row.coords[0]) + if geomtype== 'polygon': + w.poly([row.coords]) w.close()