結果 : how to do an if statement in a while loop python