Game of life python numpy