結果 : how to write prime number code in python