Pregunta de entrevista de Continental

python code for prime number