Pregunta de entrevista de Apple

write a one liner in python to do calculate the square of numbers in an array