Pregunta de entrevista de Tesla

To code a simple numpy function