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

cimarronm · cimarronm · commit ebb2e63bc22e · 2015-03-05T18:32:49.000-08:00
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -1732,9 +1732,13 @@ 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()
+            try:
+                spine = self.axes.spines['bottom']
+                spinebbox = spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()
+            except KeyError:
+                # use axes if spine doesn't exist
+                spinebbox = self.axes.bbox
             bbox = mtransforms.Bbox.union(bboxes + [spinebbox])
             bottom = bbox.y0
 
@@ -1743,9 +1747,13 @@ def _update_label_position(self, bboxes, bboxes2):
             )
 
         else:
-            spine = self.axes.spines['top']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
+            try:
+                spine = self.axes.spines['top']
+                spinebbox = spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()
+            except KeyError:
+                # use axes if spine doesn't exist
+                spinebbox = self.axes.bbox
             bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox])
             top = bbox.y1
 
@@ -2040,9 +2048,13 @@ 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()
+            try:
+                spine = self.axes.spines['left']
+                spinebbox = spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()
+            except KeyError:
+                # use axes if spine doesn't exist
+                spinebbox = self.axes.bbox
             bbox = mtransforms.Bbox.union(bboxes + [spinebbox])
             left = bbox.x0
 
@@ -2051,9 +2063,13 @@ def _update_label_position(self, bboxes, bboxes2):
             )
 
         else:
-            spine = self.axes.spines['right']
-            spinebbox = spine.get_transform().transform_path(
-                spine.get_path()).get_extents()
+            try:
+                spine = self.axes.spines['right']
+                spinebbox = spine.get_transform().transform_path(
+                    spine.get_path()).get_extents()
+            except KeyError:
+                # use axes if spine doesn't exist
+                spinebbox = self.axes.bbox
             bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox])
             right = bbox.x1
 
