game_of_life.py

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

def iterate(Z):
    # Count neighbors
    N = (
        Z[0:-2,0:-2] + Z[0:-2, 1:-1] + Z[0:-2,2:] +
        Z[1:-1,0:-2]                 + Z[1:-1,2:] +
        Z[2:  ,0:-2] + Z[2:  ,1:-1]  + Z[2:  ,2:]
    )

    # Apply rules
    birth = (N==3) & (Z[1:-1, 1:-1]==0)
    survive = ((N==2) | (N==3)) & (Z[1:-1, 1:-1]==1)
    Z[...] = 0
    Z[1:-1, 1:-1][birth | survive] = 1
    return Z

Z = np.random.randint(0,2,(300, 300))
Z[:125,:] = 0 
Z[175:,:] = 0
Z[:,:125] = 0
Z[:,175:] = 0

#Z[50:250]=0

# Set up plot
fig, ax = plt.subplots()
img = ax.imshow(Z, cmap='binary')

def update(frame):
    global Z
    Z = iterate(Z)
    img.set_data(Z)
    return [img]

ani = FuncAnimation(fig, update, frames=100000, interval=.0001, blit=True)
plt.show()