Handle plots with no spines or transformations with non-standard spines

cimarronm · cimarronm · commit f8da37b387b7 · 2015-03-05T17:39:23.000-08:00
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -1732,21 +1732,27 @@ def _update_label_position(self, bboxes, bboxes2):
             return
         x, y = self.label.get_position()
         if self.label_position == 'bottom':
-            spine = self.axes.spines['bottom']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
-            bbox = mtransforms.Bbox.union(bboxes + [spinebbox])
+            try:
+                spine = self.axes.spines['bottom']
+                spinebboxes = [spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()]
+            except KeyError:
+                spinebboxes = []
+            bbox = mtransforms.Bbox.union(bboxes + spinebboxes)
             bottom = bbox.y0
 
             self.label.set_position(
                 (x, bottom - self.labelpad * self.figure.dpi / 72.0)
             )
 
         else:
-            spine = self.axes.spines['top']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
-            bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox])
+            try:
+                spine = self.axes.spines['top']
+                spinebboxes = [spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()]
+            except KeyError:
+                spinebboxes = []
+            bbox = mtransforms.Bbox.union(bboxes2 + spinebboxes)
             top = bbox.y1
 
             self.label.set_position(
@@ -2040,21 +2046,27 @@ def _update_label_position(self, bboxes, bboxes2):
             return
         x, y = self.label.get_position()
         if self.label_position == 'left':
-            spine = self.axes.spines['left']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
-            bbox = mtransforms.Bbox.union(bboxes + [spinebbox])
+            try:
+                spine = self.axes.spines['left']
+                spinebboxes = [spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()]
+            except KeyError:
+                spinebboxes = []
+            bbox = mtransforms.Bbox.union(bboxes + spinebboxes)
             left = bbox.x0
 
             self.label.set_position(
                 (left - self.labelpad * self.figure.dpi / 72.0, y)
             )
 
         else:
-            spine = self.axes.spines['right']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
-            bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox])
+            try:
+                spine = self.axes.spines['right']
+                spinebboxes = [spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()]
+            except KeyError:
+                spinebboxes = []
+            bbox = mtransforms.Bbox.union(bboxes2 + spinebboxes)
             right = bbox.x1
 
             self.label.set_position(
