JUEGO DE LA VIDA

necesito que se complete el juego de la vida en phyton, faltarian los graficos.


import numpy as np
import random as rd
import mathplotlib.pyplot as plt

def inicial(celdasx,celdasy):
TABLERO=np.zeros([celdasx+2,celdasy+2])
for i in range (celdasx+1):
for j in range (celdasy+1):
TABLERO[i][j]=rd.randint(0,1)
return TABLERO

def principal(TABLERO,fin):
contador=0
C=[TABLERO]
numero_celdasx=len(C[0]-2)
numero_celdasy=len(C)-2

while contador<fin:
CO=np.zeros([numero_celdasx+2,numero_celdasy+2])
for i in range (numero_celdasx+1):
for j in range(numero_celdasy+1):
celula=TABLERO[i][j]
vecinos=TABLERO[i][j+1]+TABLERO[i][j-1]+TABLERO[i-1][j]+TABLERO[i+1][j]+TABLERO[i+1][j+1]+TABLERO[i+1][j-1]+TABLERO[i-1][j+1]+TABLERO[i-1][j-1]
if celula==0 and vecinos==3:
CO[i][j]=1
elif celula==1 and (vecinos==2 or vecinos==3):
CO[i][j]=1
else:
CO[i][j]=0
C.append(CO)
TABLERO=CO
contador+=1
return C

TABLERO=inicial(20,20)
C=principal(TABLERO, 20)