結果 : write code to check if a number is prime in python