結果 : code to write prime number in python