Add widths, heights and angles setter to EllipseCollection

Author: ericpre

lib/matplotlib/collections.py

@@ -1795,6 +1795,18 @@ def _set_transforms(self):
             m[:2, 2:] = 0
             self.set_transform(_affine(m))
 
+    def set_widths(self, widths):
+        self._widths = 0.5 * np.asarray(widths).ravel()
+        self.stale = True
+
+    def set_angles(self, angles):
+        self._angles = np.deg2rad(angles).ravel()
+        self.stale = True
+
+    def set_heights(self, heights):
+        self._heights = 0.5 * np.asarray(heights).ravel()
+        self.stale = True
+
     @artist.allow_rasterization
     def draw(self, renderer):
         self._set_transforms()

lib/matplotlib/tests/test_collections.py

@@ -408,6 +408,37 @@ def test_EllipseCollection():
     ax.autoscale_view()
 
 
+def test_EllipseCollection_setter():
+    # Test widths, heights and angle setter
+    rng = np.random.default_rng(0)
+
+    widths = (2, )
+    heights = (3, )
+    angles = (45, )
+    offsets = rng.random((10, 2)) * 10
+
+    fig, ax = plt.subplots()
+
+    ec = mcollections.EllipseCollection(
+        widths=widths,
+        heights=heights,
+        angles=angles,
+        offsets=offsets,
+        units='x',
+        offset_transform=ax.transData,
+        )
+
+    ax.add_collection(ec)
+    ax.set_xlim(-2, 12)
+    ax.set_ylim(-2, 12)
+
+    new_widths = rng.random((10, 2)) * 2
+    new_heights = rng.random((10, 2)) * 3
+    new_angles = rng.random((10, 2)) * 180
+
+    ec.set(widths=new_widths, heights=new_heights, angles=new_angles)
+
+
 @image_comparison(['polycollection_close.png'], remove_text=True, style='mpl20')
 def test_polycollection_close():
     from mpl_toolkits.mplot3d import Axes3D  # type: ignore