Pergunta de entrevista da empresa Continental

python code for prime number