share axes can take in an iterable axes

lkjell · lkjell · commit d31d4facae2c · 2018-03-10T01:01:17.000+01:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -4352,27 +4352,33 @@ def _share_axes_asymmetric(A, B, shared_axes):
         getattr(B, "_shared_{}_inv_axes".format(shared_axes)).add(A)
 
     def _share_axes(self, axes, shared_axes, symmetric=True, transitive=True):
-        if symmetric:
-            self._share_axes_symmetric(self, axes, shared_axes)
+        if not iterable(axes):
+            axes_iter = [axes]
         else:
-            self._share_axes_asymmetric(self, axes, shared_axes)
+            axes_iter = axes
 
-        if transitive:
-            sharedA = getattr(self, "_shared_{}_axes".format(shared_axes))
-            sharedB = getattr(axes, "_shared_{}_axes".format(shared_axes))
+        for axes in axes_iter:
+            if symmetric:
+                self._share_axes_symmetric(self, axes, shared_axes)
+            else:
+                self._share_axes_asymmetric(self, axes, shared_axes)
 
-            sharedDiffBA = sharedB - sharedA
-            sharedDiffAB = sharedA - sharedB
+            if transitive:
+                sharedA = getattr(self, "_shared_{}_axes".format(shared_axes))
+                sharedB = getattr(axes, "_shared_{}_axes".format(shared_axes))
 
-            if symmetric:
-                for ax in sharedDiffAB:
-                    self._share_axes_symmetric(axes, ax, shared_axes)
+                sharedDiffBA = sharedB - sharedA
+                sharedDiffAB = sharedA - sharedB
 
-                for ax in sharedDiffBA:
-                    self._share_axes_symmetric(self, ax, shared_axes)
-            else:
-                for ax in sharedDiffBA:
-                    self._share_axes_asymmetric(self, ax, shared_axes)
+                if symmetric:
+                    for ax in sharedDiffAB:
+                        self._share_axes_symmetric(axes, ax, shared_axes)
+
+                    for ax in sharedDiffBA:
+                        self._share_axes_symmetric(self, ax, shared_axes)
+                else:
+                    for ax in sharedDiffBA:
+                        self._share_axes_asymmetric(self, ax, shared_axes)
 
     def share_x_axes(self, axes, symmetric=True, transitive=True):
         """
