Fix: world plot limits

Author: heptaflar

doc/doc.ipynb

@@ -234,7 +234,7 @@
    "source": [
     "def plot_world(\n",
     "    data,\n",
-    "    title: str,\n",
+    "    title: str | None = None,\n",
     "    fn: str | None = None,\n",
     "    cbar_label: str | None = None,\n",
     "    cmap: str | None = None,\n",
@@ -243,6 +243,8 @@
     "    vmax=None,\n",
     "    x: str | None = None,\n",
     "    y: str | None = None,\n",
+    "    xlim: tuple | None = None,\n",
+    "    ylim: tuple | None = None,\n",
     "):\n",
     "    fig = WorldPlot().plot(\n",
     "        data=data,\n",
@@ -256,6 +258,8 @@
     "        vmax=vmax,\n",
     "        x=x,\n",
     "        y=y,\n",
+    "        xlim=xlim,\n",
+    "        ylim=ylim,\n",
     "    )\n",
     "    fig.clear()"
    ]
@@ -350,6 +354,7 @@
     "    \"DIC (10 m, µmol kg-1)\",\n",
     "    vmin=1750,\n",
     "    vmax=2250,\n",
+    "    ylim=(-8625155.13, 8625155.13),\n",
     ")"
    ]
   },
@@ -367,6 +372,7 @@
     "    \"DIC uncertainty (10 m, µmol kg-1)\",\n",
     "    vmin=0.0,\n",
     "    vmax=50.0,\n",
+    "    ylim=(-8625155.13, 8625155.13),\n",
     ")"
    ]
   },
@@ -384,6 +390,7 @@
     "    \"DIC uncertainty (10 m, µmol kg-1)\",\n",
     "    vmin=0.0,\n",
     "    vmax=50.0,\n",
+    "    ylim=(-8625155.13, 8625155.13),\n",
     ")"
    ]
   },

kaleidoscope/val/plots.py

@@ -71,7 +71,7 @@ def plot(
             cbar_kwargs=cbar_kwargs,
         )
         ax.autoscale_view()
-        self.decorate(ax, xlocs, ylocs)
+        self.decorate(ax, xlocs, ylocs, xlim, ylim)
         if title is not None:
             ax.set_title(title)
         if fn is not None:
@@ -81,7 +81,7 @@ def plot(
         plt.close()
         return fig
 
-    def decorate(self, ax, xlocs, ylocs):
+    def decorate(self, ax, xlocs, ylocs, xlim, ylim):
         """Decorates the map with land features and grid lines."""
         from cartopy.mpl.geoaxes import GeoAxes
 
@@ -95,6 +95,10 @@ def decorate(self, ax, xlocs, ylocs):
                 xlocs=xlocs,
                 ylocs=ylocs,
             )
+        if xlim is not None:
+            ax.set_xlim(xlim)
+        if ylim is not None:
+            ax.set_ylim(ylim)
 
     @property
     def land(self):