BUG: Fix relim() to support Collection artists (scatter, etc.) relim() previously skipped Collection instances entirely. Two issues: 1. add_collection() did not call collection._set_in_autoscale(True), unlike add_line(), add_patch(), and add_image(). This caused relim() to skip collections via the _get_in_autoscale() check. 2. relim() had no handling for Collection instances. Added an elif branch that calls get_datalim() and update_datalim(), mirroring the logic already used in add_collection(). Closes #30859

tinezivic · tinezivic · commit cb8bce5387aa · 2026-04-19T13:25:27.000+02:00
diff --git a/doc/api/next_api_changes/behavior/31530-TZ.rst b/doc/api/next_api_changes/behavior/31530-TZ.rst
@@ -0,0 +1,6 @@
+``relim()`` now accounts for Collection artists
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+Previously, `~.axes.Axes.relim` did not recalculate data limits for
+`.Collection` artists (e.g. those created by `~.axes.Axes.scatter`).
+Calling ``ax.relim()`` followed by ``ax.autoscale_view()`` now correctly
+includes scatter plots and other collections in the axes limits.
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2434,6 +2434,7 @@ def add_collection(self, collection, autolim=True):
                 self._request_autoscale_view()
 
         self.stale = True
+        collection._set_in_autoscale(True)
         return collection
 
     def add_image(self, image):
@@ -2638,15 +2639,11 @@ def relim(self, visible_only=False):
         """
         Recompute the data limits based on current artists.
 
-        At present, `.Collection` instances are not supported.
-
         Parameters
         ----------
         visible_only : bool, default: False
             Whether to exclude invisible artists.
         """
-        # Collections are deliberately not supported (yet); see
-        # the TODO note in artists.py.
         self.dataLim.ignore(True)
         self.dataLim.set_points(mtransforms.Bbox.null().get_points())
         self.ignore_existing_data_limits = True
@@ -2661,6 +2658,23 @@ def relim(self, visible_only=False):
                     self._update_patch_limits(artist)
                 elif isinstance(artist, mimage.AxesImage):
                     self._update_image_limits(artist)
+                elif isinstance(artist, mcoll.Collection):
+                    datalim = artist.get_datalim(self.transData)
+                    points = datalim.get_points()
+                    if not np.isinf(datalim.minpos).all():
+                        points = np.concatenate([points,
+                                                 [datalim.minpos]])
+                    x_is_data, y_is_data = (
+                        artist.get_transform()
+                        .contains_branch_separately(self.transData))
+                    ox_is_data, oy_is_data = (
+                        artist.get_offset_transform()
+                        .contains_branch_separately(self.transData))
+                    self.update_datalim(
+                        points,
+                        updatex=x_is_data or ox_is_data,
+                        updatey=y_is_data or oy_is_data,
+                    )
 
     def update_datalim(self, xys, updatex=True, updatey=True):
         """
