Add Figure.choropleth to plot choropleth maps

Author: seisman

pygmt/figure.py

@@ -517,6 +517,7 @@ def _repr_html_(self):
 
     from pygmt.src import (  # type: ignore [misc]
         basemap,
+        choropleth,
         coast,
         colorbar,
         contour,

pygmt/src/__init__.py

@@ -5,6 +5,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,23 @@
+"""
+choropleth - Plot a choropleth map.
+"""
+
+
+def choropleth(self, data, fillcol: str, **kwargs):
+    """
+    Plot a choropleth map.
+
+    Parameters
+    ----------
+    data : :class:`geopandas.GeoDataFrame`
+        The geopandas dataframe containing the geometry and data to plot.
+    fillcol
+        The column name of the data to use for the fill.
+    """
+    self.plot(
+        data=data,
+        close=True,
+        fill="+z",
+        aspatial=f"Z={fillcol}",
+        **kwargs,
+    )