結果 : write code for prime number in python