make grid spec iterable

leejjoon · leejjoon · commit f609ebd237a9 · 2010-07-31T09:18:25.000Z
svn path=/trunk/matplotlib/; revision=8606
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -128,6 +128,12 @@ def get_grid_positions(self, fig):
         return figBottoms, figTops, figLefts, figRights
 
 
+    def __iter__(self):
+        nrows, ncols = self.get_geometry()
+        total = nrows*ncols
+
+        return iter([self.__getitem__(i) for i in range(total)])
+
     def __getitem__(self, key):
         """
         create and return a SuplotSpec instance.
