Updating solutions

Author: AllenDowney

soln/Cell2D.py

@@ -51,33 +51,40 @@ def draw(self, **options):
         """Draws the array."""
         draw_array(self.array, **options)
 
-    def animate(self, frames, interval=None, step=None):
+    def animate(self, frames, interval=0.001, step=None):
         """Animate the automaton.
 
         frames: number of frames to draw
         interval: time between frames in seconds
-        iters: number of steps between frames
+        step: function to call between frames
         """
         if step is None:
             step = self.step
 
+        plt.ion()
         plt.figure()
         try:
             for _ in range(frames - 1):
                 self.draw()
-                plt.show()
-                if interval:
-                    sleep(interval)
+                plt.pause(interval)
                 step()
                 clear_output(wait=True)
             self.draw()
             plt.show()
         except KeyboardInterrupt:
             pass
+        finally:
+            plt.ioff()
 
 
 def draw_array(array, **options):
-    """Draws the cells."""
+    """Draws the cells.
+    
+    array: numpy array
+    options: passed to plt.imshow
+
+    returns: plt.imshow object
+    """
     n, m = array.shape
     options = underride(
         options,

soln/Life.py

@@ -58,7 +58,7 @@ def main(script, *args):
     life.add_cells(n // 2 + 12, col, *lwss)
     life.add_cells(n // 2 + 26, col, *lwss)
     life.add_cells(n // 2 + 19, col, *bhep)
-    life.animate(frames=100)
+    life.animate(frames=1000)
     plt.subplots_adjust(left=0.01, right=0.99, bottom=0.01, top=0.99)
     plt.show()
 

soln/LifeRabbits.py

@@ -25,7 +25,7 @@ def main(script, *args):
     m = 600
     life = Life(n, m)
     life.add_cells(n // 2, m // 2, *rabbits)
-    life.animate(frames=100, interval=1)
+    life.animate(frames=1000)
     plt.subplots_adjust(left=0.01, right=0.99, bottom=0.01, top=0.99)
     plt.show()