Pregunta de entrevista de Capco

Basic Python array code test