Add Figure.choropleth to plot choropleth maps

Author: seisman

pygmt/figure.py

@@ -520,6 +520,7 @@ def _repr_html_(self):
 
     from pygmt.src import (  # pylint: disable=import-outside-toplevel
         basemap,
+        choropleth,
         coast,
         colorbar,
         contour,

pygmt/src/__init__.py

@@ -6,6 +6,7 @@
 from pygmt.src.basemap import basemap
 from pygmt.src.binstats import binstats
 from pygmt.src.blockm import blockmean, blockmedian, blockmode
+from pygmt.src.choropleth import choropleth
 from pygmt.src.coast import coast
 from pygmt.src.colorbar import colorbar
 from pygmt.src.config import config

pygmt/src/choropleth.py

@@ -0,0 +1,20 @@
+def choropleth(self, data, fillcol, cmap=True, **kwargs):
+    """
+    Plot a choropleth map from a :class:`geopandas.GeoDataFrame` object.
+
+    Parameters
+    ----------
+    data : :class:`geopandas.GeoDataFrame`
+        The geopandas dataframe containing the geometry and data to plot.
+    fillcol : str
+        The column name of the data to use for the fill.
+    cmap : str
+    """
+    self.plot(
+        data=data,
+        close=True,
+        fill="+z",
+        cmap=cmap,
+        aspatial=f"Z={fillcol}",
+        **kwargs,
+    )