結果 : how to put an if statement in a for loop python