結果 : write a prime number code in python