Merge pull request #540 from mdboom/issue540

efiring · efiring · commit bdc427ab9c45 · 2011-12-31T15:32:19.000-08:00
Raise a useful error message when there is no current image
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -1682,6 +1682,8 @@ def colormaps():
 def colorbar(mappable=None, cax=None, ax=None, **kw):
     if mappable is None:
         mappable = gci()
+        if mappable is None:
+            raise RuntimeError('You must first define an image, eg with imshow')
     if ax is None:
         ax = gca()
 
@@ -1732,6 +1734,9 @@ def set_cmap(cmap):
 
     if im is not None:
         im.set_cmap(cmap)
+    else:
+        raise RuntimeError('You must first define an image, eg with imshow')
+
     draw_if_interactive()
 
 
