Comentários?
def matriz_vizinhos(nlins, ncols): viz = 4 * np.ones( (nlins,ncols), int) viz[1:-1,1] = viz[1:-1,-2] = viz[1,1:-1] = viz[-2,1:-1] = 3 viz[1,1] = viz[1,-2] = viz[-2,1] = viz[-2,-2] = 2 viz[:,0] = viz[:,-1] = viz[0,:] = viz[-1,:] = 0 return viz