added plotting of the canvas

Author: anuragkumarak95

game_o_life.py

@@ -31,9 +31,13 @@
 import random, time, sys
 from matplotlib import pyplot as plt
 import matplotlib.animation as animation
+from matplotlib.colors import ListedColormap
 
 usage_doc='Usage of script: script_nama <size_of_canvas:int>'
 
+choice = [0]*100 + [1]*10
+random.shuffle(choice)
+
 def create_canvas(size):
     canvas = [ [False for i in range(size)] for j in range(size)]
     return canvas
@@ -102,7 +106,13 @@ def __judge_point(pt,neighbours):
     seed(c)
     fig, ax = plt.subplots()
     fig.show() 
-    while True:
-        c = run(c)
-        ax.matshow(c)
-        fig.canvas.draw()
+    cmap = ListedColormap(['w','k'])
+    try:
+        while True:
+            c = run(c)            
+            ax.matshow(c,cmap=cmap)
+            fig.canvas.draw()
+            ax.cla() 
+    except KeyboardInterrupt:
+        # do nothing.
+        pass